From 958a17199ac519504e390ea0d53295ceb8cbd2c1 Mon Sep 17 00:00:00 2001
From: comfyanonymous <121283862+comfyanonymous@users.noreply.github.com>
Date: Mon, 3 Nov 2025 14:08:30 -0800
Subject: [PATCH] People should update their pytorch versions. (#10618)

---
 comfy/quant_ops.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/comfy/quant_ops.py b/comfy/quant_ops.py
index 873f173ed..5af6f118e 100644
--- 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]))