People should update their pytorch versions. (#10618)

2025-11-03 14:08:30 -08:00 · 2025-11-03 14:08:30 -08:00 · 958a17199a
parent e974e554ca
commit 958a17199a
1 changed files with 4 additions and 0 deletions
--- a/comfy/quant_ops.py
+++ b/comfy/quant_ops.py
@ -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]))