youngkingdom/ComfyUI-WanMoeKSampler
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

update shift logic

Browse Source
This commit is contained in:
Stéphane du Hamel
2025-08-12 13:18:49 +02:00
parent 12cfa70738
commit 57987b58aa
1 changed files with 7 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
nodes.py
View File

@ -68,17 +68,16 @@ def wan_ksampler(model_high_noise, model_low_noise, seed, steps, cfgs, sampler_n
return (out, ) return (out, )
def set_shift(model,sigma_shift): def set_shift(model,sigma_shift):
model_sampling = model.get_model_object("model_sampling")
sampling_base = comfy.model_sampling.ModelSamplingDiscreteFlow if not model_sampling:
sampling_type = comfy.model_sampling.CONST sampling_base = comfy.model_sampling.ModelSamplingDiscreteFlow
sampling_type = comfy.model_sampling.CONST
class ModelSamplingAdvanced(sampling_base, sampling_type):
pass
class ModelSamplingAdvanced(sampling_base, sampling_type): model_sampling = ModelSamplingAdvanced(model.model.model_config)
pass
model_sampling = ModelSamplingAdvanced(model.model.model_config)
model_sampling.set_parameters(shift=sigma_shift, multiplier=1000) model_sampling.set_parameters(shift=sigma_shift, multiplier=1000)
model.add_object_patch("model_sampling", model_sampling) model.add_object_patch("model_sampling", model_sampling)
model.add_object_patch("model_sampling", model_sampling)
return model return model
class WanMoeKSampler: class WanMoeKSampler: