Fix ConditioningZeroOut when there is no pooled output.

c3db3447 · comfyanonymous · d1600738 · c3db3447
Commit c3db3447 authored Jul 09, 2024 by comfyanonymous
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 2 deletions

nodes.py nodes.py +3 -2

No files found.
--- a/nodes.py
+++ b/nodes.py
@@ -232,8 +232,9 @@ class ConditioningZeroOut:
        c = []
        for t in conditioning:
            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]
            c.append(n)
        return (c, )