fix issue with ddim and normalization of lowres conditioning image

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

View File

@@ -2328,6 +2328,9 @@ class Decoder(nn.Module):
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'):
alpha = alphas[time]
alpha_next = alphas[time_next]