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

allow tcfg node to skip shorter unconditional prediction

Browse Source
This commit is contained in:
Extraltodeus 2025-08-21 03:43:17 +02:00 committed by GitHub
parent e73a9dbe30
commit aef957683c
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
comfy_extras/nodes_tcfg.py
View File

@ -49,7 +49,7 @@ class TCFG(ComfyNodeABC):
# Assume [cond, uncond, ...] # Assume [cond, uncond, ...]
x = args["input"] x = args["input"]
conds_out = args["conds_out"] conds_out = args["conds_out"]
if len(conds_out) <= 1 or None in args["conds"][:2]: if len(conds_out) <= 1 or None in args["conds"][:2] or not args["conds_out"][1].any():
# Skip when either cond or uncond is None # Skip when either cond or uncond is None
return conds_out return conds_out
cond_pred = conds_out[0] cond_pred = conds_out[0]