Fix regression. (#13053)

This commit is contained in:
comfyanonymous
2026-03-18 15:35:25 -07:00
committed by GitHub
parent 9fff091f35
commit 56ff88f951

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,)