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

address https://github.com/lucidrains/DALLE2-pytorch/issues/266

Browse Source
This commit is contained in:
Phil Wang
2022-11-23 08:41:18 -08:00
parent 91c8d1ca13
commit 067ac323da
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
dalle2_pytorch/dalle2_pytorch.py
View File

@@ -1320,7 +1320,7 @@ class DiffusionPrior(nn.Module):
elif self.predict_x_start:
x_start = pred
else:
x_start = self.noise_scheduler.predict_start_from_noise(image_embed, t = time_cond, noise = pred_noise)
x_start = self.noise_scheduler.predict_start_from_noise(image_embed, t = time_cond, noise = pred)
# clip x0 before maybe predicting noise

2
dalle2_pytorch/version.py
View File

@@ -1 +1 @@
__version__ = '1.11.1'
__version__ = '1.11.2'