From 135d3025ea0bf5b65ebd78558503f9237683c233 Mon Sep 17 00:00:00 2001 From: lspindler Date: Tue, 28 Oct 2025 07:33:42 +0100 Subject: [PATCH] Rename quant dtype parameter --- tests-unit/comfy_quant/test_quant_registry.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests-unit/comfy_quant/test_quant_registry.py b/tests-unit/comfy_quant/test_quant_registry.py index 26e91a7ee..2d7d3fa28 100644 --- a/tests-unit/comfy_quant/test_quant_registry.py +++ b/tests-unit/comfy_quant/test_quant_registry.py @@ -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)