xinyun/ComfyUI-KJNodes
1
0
Fork 0
mirror of https://git.datalinker.icu/kijai/ComfyUI-KJNodes.git synced 2025-12-10 05:15:05 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update nodes.py

Browse Source
This commit is contained in:
kijai 2025-02-24 19:26:34 +02:00
parent 69ec71d2bd
commit 4b3009e4bf
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
nodes/nodes.py
View File

@ -2442,6 +2442,7 @@ class Guider_ScheduledCFG(CFGGuider):
def predict_noise(self, x, timestep, model_options={}, seed=None):
steps = model_options["transformer_options"]["sample_sigmas"]
matched_step_index = (steps == timestep).nonzero()
assert not (isinstance(self.cfg, list) and len(self.cfg) != (len(steps) - 1)), "cfg list length must match step count"
if len(matched_step_index) > 0:
current_step_index = matched_step_index.item()
else:
@ -2463,6 +2464,7 @@ class Guider_ScheduledCFG(CFGGuider):
else:
uncond = None
cfg = 1.0
return sampling_function(self.inner_model, x, timestep, uncond, self.conds.get("positive", None), cfg, model_options=model_options, seed=seed)
class ScheduledCFGGuidance: