aljaz/DALLE2-pytorch
1
1
Fork 0
mirror of https://github.com/lucidrains/DALLE2-pytorch.git synced 2025-12-19 09:44:19 +01:00
Code Issues Packages Projects Releases Wiki Activity

fix misnamed variable, thanks to @nousr

Browse Source
This commit is contained in:
Phil Wang
2022-07-09 19:01:37 -07:00
parent a598820012
commit 3dae43fa0e
2 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
dalle2_pytorch/dalle2_pytorch.py
View File

@@ -1061,9 +1061,9 @@ class DiffusionPrior(nn.Module):
c2 = ((1 - alpha_next) - torch.square(c1)).sqrt() c2 = ((1 - alpha_next) - torch.square(c1)).sqrt()
noise = torch.randn_like(image_embed) if time_next > 0 else 0. noise = torch.randn_like(image_embed) if time_next > 0 else 0.
img = x_start * alpha_next.sqrt() + \ image_embed = x_start * alpha_next.sqrt() + \
c1 * noise + \ c1 * noise + \
c2 * pred_noise c2 * pred_noise
return image_embed return image_embed

2
dalle2_pytorch/version.py
View File

@@ -1 +1 @@
__version__ = '0.19.4' __version__ = '0.19.5'