wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-02-16 08:22:36 +08:00
Code Issues Actions 15 Packages Projects Releases Wiki Activity

Rename quant dtype parameter

Browse Source
This commit is contained in:
lspindler 2025-10-28 07:33:42 +01:00
parent 59a2e8c74e
commit 135d3025ea
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tests-unit/comfy_quant/test_quant_registry.py
View File

@ -122,7 +122,7 @@ class TestTensorCoreFP8Layout(unittest.TestCase):
qdata, layout_params = TensorCoreFP8Layout.quantize(
float_tensor,
scale=scale,
fp8_dtype=torch.float8_e4m3fn
dtype=torch.float8_e4m3fn
)
self.assertEqual(qdata.dtype, torch.float8_e4m3fn)
@ -139,7 +139,7 @@ class TestTensorCoreFP8Layout(unittest.TestCase):
qdata, layout_params = TensorCoreFP8Layout.quantize(
float_tensor,
scale=scale,
fp8_dtype=torch.float8_e4m3fn
dtype=torch.float8_e4m3fn
)
dequantized = TensorCoreFP8Layout.dequantize(qdata, **layout_params)