diff --git a/ldm/models/diffusion/ddpm.py b/ldm/models/diffusion/ddpm.py index 4e7f5c3..8c648fe 100644 --- a/ldm/models/diffusion/ddpm.py +++ b/ldm/models/diffusion/ddpm.py @@ -1555,7 +1555,7 @@ class LatentUpscaleDiffusion(LatentDiffusion): uc[k] = [uc_tmp] elif k == "c_adm": assert isinstance(c[k], torch.Tensor) - uc[k] = torch.ones_like(c[k]) * (self.low_scale_model.max_max_noise_level-1) + uc[k] = torch.ones_like(c[k]) * (self.low_scale_model.max_noise_level-1) elif isinstance(c[k], list): uc[k] = [torch.zeros_like(c[k][i]) for i in range(len(c[k]))] else: