wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-03-24 18:43:36 +08:00
Code Issues Actions 15 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]:
v = o[2][k]
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
r = r + (extra,)