xinyun/ComfyUI
1
0
Fork 0
mirror of https://git.datalinker.icu/comfyanonymous/ComfyUI synced 2026-09-19 02:24:27 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update nodes.py

Browse Source
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
1 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
nodes.py
View File

@ -55,9 +55,8 @@ class CLIPTextEncode:
def encode(self, clip, text): def encode(self, clip, text):
tokens = clip.tokenize(text) tokens = clip.tokenize(text)
output = clip.encode_from_tokens(tokens, return_pooled=True, return_dict=True) cond, pooled = clip.encode_from_tokens(tokens, return_pooled=True)
cond = output.pop("cond") return ([[cond, {"pooled_output": pooled}]], )
return ([[cond, output]], )
class ConditioningCombine: class ConditioningCombine:
@classmethod @classmethod