Update nodes.py

This commit is contained in:
DivineSpring 2024-07-14 15:37:09 +03:00 committed by GitHub
parent 4095872d6e
commit a4a36cb1c7
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -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