diff --git a/comfy_extras/nodes_tcfg.py b/comfy_extras/nodes_tcfg.py index 1a6767770..118e79ba5 100644 --- a/comfy_extras/nodes_tcfg.py +++ b/comfy_extras/nodes_tcfg.py @@ -51,7 +51,7 @@ class TCFG(io.ComfyNode): # Assume [cond, uncond, ...] x = args["input"] 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 return conds_out cond_pred = conds_out[0]