wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-05-15 19:47:24 +08:00
Code Issues Actions 15 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
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 1 deletions
Show Stats 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)