From a4a36cb1c796ca2deaaf0d667d03edc30cc411e4 Mon Sep 17 00:00:00 2001
From: DivineSpring <46858816+MaayanBogin@users.noreply.github.com>
Date: Sun, 14 Jul 2024 15:37:09 +0300
Subject: [PATCH] Update nodes.py

---
 nodes.py | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/nodes.py b/nodes.py
index 5778f0609..8d8b153f8 100644
--- a/nodes.py
+++ b/nodes.py
@@ -55,9 +55,8 @@ class CLIPTextEncode:
 
     def encode(self, clip, text):
         tokens = clip.tokenize(text)
-        output = clip.encode_from_tokens(tokens, return_pooled=True, return_dict=True)
-        cond = output.pop("cond")
-        return ([[cond, output]], )
+        cond, pooled = clip.encode_from_tokens(tokens, return_pooled=True)
+        return ([[cond, {"pooled_output": pooled}]], )
 
 class ConditioningCombine:
     @classmethod