This commit is contained in:
Patrick Esser 2022-06-14 14:52:41 +00:00 committed by root
parent a0f40fb5ff
commit 2d7af412c1

View file

@ -1555,7 +1555,7 @@ class LatentUpscaleDiffusion(LatentDiffusion):
uc[k] = [uc_tmp] uc[k] = [uc_tmp]
elif k == "c_adm": elif k == "c_adm":
assert isinstance(c[k], torch.Tensor) 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): elif isinstance(c[k], list):
uc[k] = [torch.zeros_like(c[k][i]) for i in range(len(c[k]))] uc[k] = [torch.zeros_like(c[k][i]) for i in range(len(c[k]))]
else: else: