r/stable-diffusion-finetune
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

solve interpretability

Browse source
This commit is contained in:
Patrick Esser 2022-06-13 10:56:17 +00:00 committed by root
parent 3c77204c0a
commit 046410167f
1 changed files with 3 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
ldm/models/diffusion/ddpm.py
View file

@ -1476,6 +1476,9 @@ class LatentUpscaleDiffusion(LatentDiffusion):
all_conds = {"c_concat": [zx], "c_crossattn": [c], "c_adm": noise_level}
if log_mode:
# TODO: maybe disable if too expensive
interpretability = True
if interpretability:
zx = zx[:, :, ::2, ::2]
x_low_rec = self.low_scale_model.decode(zx)
return z, all_conds, x, xrec, xc, x_low, x_low_rec, noise_level
return z, all_conds