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

Persists ModelNoiseScale when also patching shift

Browse Source
This commit is contained in:
kijai 2026-05-14 15:59:33 +03:00
parent 8505abf52e
commit 6dd554add0
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): class ModelSamplingAdvanced(sampling_base, sampling_type):
pass pass
original = m.get_model_object("model_sampling")
model_sampling = ModelSamplingAdvanced(model.model.model_config) model_sampling = ModelSamplingAdvanced(model.model.model_config)
model_sampling.set_parameters(shift=shift, multiplier=multiplier) 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) m.add_object_patch("model_sampling", model_sampling)
return (m, ) return (m, )
@ -315,7 +318,7 @@ class ModelNoiseScale:
def patch(self, model, noise_scale): def patch(self, model, noise_scale):
m = model.clone() m = model.clone()
original = m.model.model_sampling original = m.get_model_object("model_sampling")
ms = type(original)(m.model.model_config) ms = type(original)(m.model.model_config)
ms.set_parameters(shift=original.shift, multiplier=original.multiplier) ms.set_parameters(shift=original.shift, multiplier=original.multiplier)
ms.set_noise_scale(noise_scale) ms.set_noise_scale(noise_scale)