Skip to content

Commit 56ff88f

Browse files
Fix regression. (Comfy-Org#13053)
1 parent 9fff091 commit 56ff88f

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

comfy/sd1_clip.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -72,7 +72,7 @@ def encode_token_weights(self, token_weight_pairs):
7272
for k in o[2]:
7373
v = o[2][k]
7474
if k == "attention_mask":
75-
v = v[:sections].flatten().unsqueeze(dim=0).to(device=model_management.intermediate_device(), dtype=model_management.intermediate_dtype())
75+
v = v[:sections].flatten().unsqueeze(dim=0).to(device=model_management.intermediate_device())
7676
extra[k] = v
7777

7878
r = r + (extra,)

0 commit comments

Comments
 (0)