typo
This commit is contained in:
parent
a0f40fb5ff
commit
2d7af412c1
1 changed files with 1 additions and 1 deletions
|
@ -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:
|
||||||
|
|
Loading…
Reference in a new issue