wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-02-14 23:42:35 +08:00
Code Issues Actions 14 Packages Projects Releases Wiki Activity

Fix recursion.

Browse Source
This commit is contained in:
comfyanonymous 2025-11-03 19:15:35 -05:00 committed by GitHub
parent 8230919053
commit 34c79a1d38
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 9 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
comfy/quant_ops.py
View File

@ -474,7 +474,15 @@ def fp8_addmm(func, args, kwargs):
output = output[0] output = output[0]
return output return output
return func(*args, **kwargs) a = list(args)
if isinstance(args[0], QuantizedTensor):
a[0] = args[0].dequantize()
if isinstance(args[1], QuantizedTensor):
a[1] = args[1].dequantize()
if isinstance(args[2], QuantizedTensor):
a[2] = args[2].dequantize()
return func(*a, **kwargs)
@register_layout_op(torch.ops.aten.view.default, "TensorCoreFP8Layout") @register_layout_op(torch.ops.aten.view.default, "TensorCoreFP8Layout")
@register_layout_op(torch.ops.aten.t.default, "TensorCoreFP8Layout") @register_layout_op(torch.ops.aten.t.default, "TensorCoreFP8Layout")