qnloft
/
ComfyUI
mirror of https://github.com/comfyanonymous/ComfyUI.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fix ConditioningZeroOut when there is no pooled output.

This commit is contained in:
comfyanonymous 2024-07-09 11:52:31 -04:00
parent d160073829
commit c3db344746
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
nodes.py
View File

@ -232,8 +232,9 @@ class ConditioningZeroOut:
c = [] c = []
for t in conditioning: for t in conditioning:
d = t[1].copy() d = t[1].copy()
if "pooled_output" in d: pooled_output = d.get("pooled_output", None)
d["pooled_output"] = torch.zeros_like(d["pooled_output"]) if pooled_output is not None:
d["pooled_output"] = torch.zeros_like(pooled_output)
n = [torch.zeros_like(t[0]), d] n = [torch.zeros_like(t[0]), d]
c.append(n) c.append(n)
return (c, ) return (c, )