xinyun/ComfyUI
1
0
Fork 0
mirror of https://git.datalinker.icu/comfyanonymous/ComfyUI synced 2025-12-09 22:14:34 +08:00
Code Issues Packages Projects Releases Wiki Activity

People should update their pytorch versions. (#10618)

Browse Source
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
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
comfy/quant_ops.py
View File

@ -418,6 +418,10 @@ def fp8_linear(func, args, kwargs):
scale_b=scale_b, scale_b=scale_b,
out_dtype=out_dtype, 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: if not tensor_2d:
output = output.reshape((-1, input_shape[1], weight.shape[0])) output = output.reshape((-1, input_shape[1], weight.shape[0]))