From 6dd554add081a959a08dc7b13bc3ba0af308d4bf Mon Sep 17 00:00:00 2001 From: kijai Date: Thu, 14 May 2026 15:59:33 +0300 Subject: [PATCH] Persists ModelNoiseScale when also patching shift --- 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)