wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-03-25 02:53:29 +08:00
Code Issues Actions 14 Packages Projects Releases Wiki Activity

Fix regression. (#13053)

Browse Source
This commit is contained in:
comfyanonymous 2026-03-18 15:35:25 -07:00 committed by GitHub
parent 9fff091f35
commit 56ff88f951
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
comfy/sd1_clip.py
View File

@ -72,7 +72,7 @@ class ClipTokenWeightEncoder:
for k in o[2]: for k in o[2]:
v = o[2][k] v = o[2][k]
if k == "attention_mask": if k == "attention_mask":
v = v[:sections].flatten().unsqueeze(dim=0).to(device=model_management.intermediate_device(), dtype=model_management.intermediate_dtype()) v = v[:sections].flatten().unsqueeze(dim=0).to(device=model_management.intermediate_device())
extra[k] = v extra[k] = v
r = r + (extra,) r = r + (extra,)