inpaint and txt2img working with ddim sampler

This commit is contained in:
Lincoln Stein
2022-10-25 10:00:28 -04:00
parent 175c7bddfc
commit aaf7a4f1d3
4 changed files with 21 additions and 8 deletions

View File

@ -2157,7 +2157,6 @@ class DiffusionWrapper(pl.LightningModule):
]
def forward(self, x, t, c_concat: list = None, c_crossattn: list = None):
print(f'DEBUG (ddpm) c_concat = {c_concat}')
if self.conditioning_key is None:
out = self.diffusion_model(x, t)
elif self.conditioning_key == 'concat':