People should update their pytorch versions. (#10618)

This commit is contained in:
comfyanonymous 2025-11-03 14:08:30 -08:00 committed by GitHub
parent e974e554ca
commit 958a17199a
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -418,6 +418,10 @@ def fp8_linear(func, args, kwargs):
scale_b=scale_b,
out_dtype=out_dtype,
)
if isinstance(output, tuple): # TODO: remove when we drop support for torch 2.4
output = output[0]
if not tensor_2d:
output = output.reshape((-1, input_shape[1], weight.shape[0]))