From a0e3b71fab5034ee1814fe1228f5d05f1adb473f Mon Sep 17 00:00:00 2001 From: lllyasviel Date: Thu, 25 Jan 2024 21:01:30 -0800 Subject: [PATCH] i --- modules/sd_samplers_cfg_denoiser.py | 2 ++ modules_forge/forge_loader.py | 2 +- 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/modules/sd_samplers_cfg_denoiser.py b/modules/sd_samplers_cfg_denoiser.py index f5f7a815..21d5006e 100644 --- a/modules/sd_samplers_cfg_denoiser.py +++ b/modules/sd_samplers_cfg_denoiser.py @@ -218,6 +218,8 @@ class CFGDenoiser(torch.nn.Module): tensor = tensor.to(unet_dtype) uncond = uncond.to(unet_dtype) + self.inner_model.inner_model.current_sigmas = sigma_in + if tensor.shape[1] == uncond.shape[1] or skip_uncond: if is_edit_model: cond_in = catenate_conds([tensor, uncond, uncond]) diff --git a/modules_forge/forge_loader.py b/modules_forge/forge_loader.py index 14ea417a..e68793fc 100644 --- a/modules_forge/forge_loader.py +++ b/modules_forge/forge_loader.py @@ -252,7 +252,7 @@ def load_model_for_a1111(timer, checkpoint_info=None, state_dict=None): def forge_unet_forward(*args, **kwargs): current_transformer_options = kwargs.get('transformer_options', {}) - current_transformer_options.update(dict(cond_or_uncond=sd_model.cond_or_uncond)) + current_transformer_options.update(dict(cond_or_uncond=sd_model.cond_or_uncond, sigmas=sd_model.current_sigmas)) current_transformer_options.update(sd_model.forge_objects.unet.model_options.get('transformer_options', {})) kwargs.update(dict(