diff --git a/modules/sd_samplers_cfg_denoiser.py b/modules/sd_samplers_cfg_denoiser.py index 09140a24..3ab72bf5 100644 --- a/modules/sd_samplers_cfg_denoiser.py +++ b/modules/sd_samplers_cfg_denoiser.py @@ -66,6 +66,10 @@ class CFGDenoiser(torch.nn.Module): raise NotImplementedError() def combine_denoised(self, x_out, conds_list, uncond, cond_scale): + if len(conds_list) == 1: + model_options = self.inner_model.inner_model.forge_objects.unet.model_options + a = 0 + denoised_uncond = x_out[-uncond.shape[0]:] denoised = torch.clone(denoised_uncond)