From 3c1ef48fe24c027131b7c846970bec963b6c2b69 Mon Sep 17 00:00:00 2001 From: Lincoln Stein Date: Tue, 25 Oct 2022 13:57:42 -0400 Subject: [PATCH] fix crash when doing img2img with ddim sampler and SD 1.5 --- ldm/models/diffusion/sampler.py | 14 -------------- 1 file changed, 14 deletions(-) diff --git a/ldm/models/diffusion/sampler.py b/ldm/models/diffusion/sampler.py index 4e9fce0102..ccebed3bfc 100644 --- a/ldm/models/diffusion/sampler.py +++ b/ldm/models/diffusion/sampler.py @@ -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