From b2000029c8290207720a95c0aff5b71b2c80d91f Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Jukka=20Sepp=C3=A4nen?= <40791699+kijai@users.noreply.github.com> Date: Fri, 15 May 2026 04:36:17 +0300 Subject: [PATCH] Persists ModelNoiseScale when also patching shift (#13892) --- comfy_extras/nodes_model_advanced.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/comfy_extras/nodes_model_advanced.py b/comfy_extras/nodes_model_advanced.py index 33b940a0f..b27ac1296 100644 --- a/comfy_extras/nodes_model_advanced.py +++ b/comfy_extras/nodes_model_advanced.py @@ -134,8 +134,11 @@ class ModelSamplingSD3: class ModelSamplingAdvanced(sampling_base, sampling_type): pass + original = m.get_model_object("model_sampling") model_sampling = ModelSamplingAdvanced(model.model.model_config) model_sampling.set_parameters(shift=shift, multiplier=multiplier) + if hasattr(original, "noise_scale"): + model_sampling.set_noise_scale(original.noise_scale) m.add_object_patch("model_sampling", model_sampling) return (m, ) @@ -315,7 +318,7 @@ class ModelNoiseScale: def patch(self, model, noise_scale): m = model.clone() - original = m.model.model_sampling + original = m.get_model_object("model_sampling") ms = type(original)(m.model.model_config) ms.set_parameters(shift=original.shift, multiplier=original.multiplier) ms.set_noise_scale(noise_scale)