Skip to content

Commit 958a171

Browse files
People should update their pytorch versions. (Comfy-Org#10618)
1 parent e974e55 commit 958a171

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

comfy/quant_ops.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -418,6 +418,10 @@ def fp8_linear(func, args, kwargs):
418418
scale_b=scale_b,
419419
out_dtype=out_dtype,
420420
)
421+
422+
if isinstance(output, tuple): # TODO: remove when we drop support for torch 2.4
423+
output = output[0]
424+
421425
if not tensor_2d:
422426
output = output.reshape((-1, input_shape[1], weight.shape[0]))
423427

0 commit comments

Comments
 (0)