From 8135fac2d1dde0492c3c17c760e670e12c5e3f2d Mon Sep 17 00:00:00 2001 From: lllyasviel Date: Thu, 25 Jan 2024 19:19:46 -0800 Subject: [PATCH] Update sd_samplers_cfg_denoiser.py --- modules/sd_samplers_cfg_denoiser.py | 4 ++++ 1 file changed, 4 insertions(+) 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)