From 9062eecebbb1d2fdd5e2a717c059dbea30d5fa2a Mon Sep 17 00:00:00 2001 From: Vyacheslav Moskalev Date: Thu, 11 Apr 2024 02:59:40 +0700 Subject: [PATCH] Fix denoiser error. --- efficiency_nodes.py | 2 +- py/smZ_cfg_denoiser.py | 1 - 2 files changed, 1 insertion(+), 2 deletions(-) diff --git a/efficiency_nodes.py b/efficiency_nodes.py index 076d758..c2804ed 100644 --- a/efficiency_nodes.py +++ b/efficiency_nodes.py @@ -480,7 +480,7 @@ def process_latent_image(model, seed, steps, cfg, sampler_name, scheduler, posit rng_source, cfg_denoiser, add_seed_noise, m_seed, m_weight = script["noise"] smZ_rng_source.rng_rand_source(rng_source) # this function monkey patches comfy.sample.prepare_noise if cfg_denoiser: - comfy.samplers.KSampler = smZ_cfg_denoiser.SDKSampler + comfy.samplers.KSampler = smZ_cfg_denoiser.KSampler if add_seed_noise: comfy.sample.prepare_noise = cg_mixed_seed_noise.get_mixed_noise_function(comfy.sample.prepare_noise, m_seed, m_weight) else: diff --git a/py/smZ_cfg_denoiser.py b/py/smZ_cfg_denoiser.py index 2fc12b4..fa52bcb 100644 --- a/py/smZ_cfg_denoiser.py +++ b/py/smZ_cfg_denoiser.py @@ -768,7 +768,6 @@ def hook_for_settings_node_and_sampling(): _Sampler = comfy.samplers.Sampler _max_denoise = comfy.samplers.Sampler.max_denoise _sample = comfy.samplers.sample - _wrap_model = comfy.samplers.wrap_model def get_value_from_args(args, kwargs, key_to_lookup, fn, idx=None): value = None