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

fix a bug with ddim and predict x0 objective

Browse Source
This commit is contained in:
Phil Wang
2022-07-18 19:04:21 -07:00
parent 3676a8ce78
commit d88c7ba56c
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

@@ -522,7 +522,7 @@ class NoiseScheduler(nn.Module):
def predict_noise_from_start(self, x_t, t, x0): def predict_noise_from_start(self, x_t, t, x0):
return ( return (
(x0 - extract(self.sqrt_recip_alphas_cumprod, t, x_t.shape) * x_t) / \ (extract(self.sqrt_recip_alphas_cumprod, t, x_t.shape) * x_t - x0) / \
extract(self.sqrt_recipm1_alphas_cumprod, t, x_t.shape) extract(self.sqrt_recipm1_alphas_cumprod, t, x_t.shape)
) )

2
dalle2_pytorch/version.py
View File

@@ -1 +1 @@
__version__ = '0.25.1' __version__ = '0.25.2'