JaehaerysNL/StableDiffusion
1
0
Fork 0
mirror of https://github.com/Stability-AI/stablediffusion.git synced 2024-12-22 15:44:58 +00:00
Code Issues Projects Releases Packages Wiki Activity

Merge pull request #64 from eltociear/patch-1

Browse source 
Fix typo in ddpm.py
This commit is contained in:
Robin Rombach 2022-12-07 13:13:19 +01:00 committed by GitHub
commit f547c4acc8
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
ldm/models/diffusion/ddpm.py
View file

@ -390,7 +390,7 @@ class DDPM(pl.LightningModule):
elif self.parameterization == "v": elif self.parameterization == "v":
target = self.get_v(x_start, noise, t) target = self.get_v(x_start, noise, t)
else: else:
raise NotImplementedError(f"Paramterization {self.parameterization} not yet supported") raise NotImplementedError(f"Parameterization {self.parameterization} not yet supported")
loss = self.get_loss(model_out, target, mean=False).mean(dim=[1, 2, 3]) loss = self.get_loss(model_out, target, mean=False).mean(dim=[1, 2, 3])