mirror of
https://git.datalinker.icu/comfyanonymous/ComfyUI
synced 2025-12-08 21:44:33 +08:00
People should update their pytorch versions. (#10618)
This commit is contained in:
parent
e974e554ca
commit
958a17199a
@ -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]))
|
||||
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user