Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix crash when doing img2img with ddim sampler and SD 1.5

Browse Source
This commit is contained in:
Lincoln Stein 2022-10-25 13:57:42 -04:00
parent c732fd0740
commit 3c1ef48fe2
1 changed files with 0 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
ldm/models/diffusion/sampler.py
View File

@ -319,20 +319,6 @@ class Sampler(object):
init_latent = None,
mask = None,
):
print(f'DEBUG(sampler): cond = {cond}')
if cond is not None:
if isinstance(cond, dict):
ctmp = cond[list(cond.keys())[0]]
while isinstance(ctmp, list):
ctmp = ctmp[0]
cbs = ctmp.shape[0]
if cbs != batch_size:
print(f"Warning: Got {cbs} conds but batch-size is {batch_size}")
else:
if cond.shape[0] != batch_size:
print(f"Warning: Got {cond.shape[0]} conditionings but batch-size is {batch_size}")
timesteps = (
np.arange(self.ddpm_num_timesteps)
if use_original_steps