comfyanonymous/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-05-26 07:58:25 +00:00
Code Issues Packages Projects Releases Wiki Activity

Persists ModelNoiseScale when also patching shift (#13892)

Browse Source
This commit is contained in:
Jukka Seppänen
2026-05-15 04:36:17 +03:00
committed by GitHub
parent b112f68681
commit b2000029c8
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
comfy_extras/nodes_model_advanced.py
View File

@@ -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)