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 issue with ddim and normalization of lowres conditioning image

Browse Source
This commit is contained in:
Phil Wang
2022-07-13 09:48:40 -07:00
parent 79e2a3bc77
commit 972ee973bc
2 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
dalle2_pytorch/dalle2_pytorch.py
View File

@@ -2328,6 +2328,9 @@ class Decoder(nn.Module):
img = torch.randn(shape, device = device) img = torch.randn(shape, device = device)
if not is_latent_diffusion:
lowres_cond_img = maybe(self.normalize_img)(lowres_cond_img)
for time, time_next in tqdm(time_pairs, desc = 'sampling loop time step'): for time, time_next in tqdm(time_pairs, desc = 'sampling loop time step'):
alpha = alphas[time] alpha = alphas[time]
alpha_next = alphas[time_next] alpha_next = alphas[time_next]

2
dalle2_pytorch/version.py
View File

@@ -1 +1 @@
__version__ = '0.23.3' __version__ = '0.23.4'