missed a dtype

syedshazli · syedshazli · commit c5857242fd6e · 2025-12-18T12:37:20.000-05:00
diff --git a/torchao/quantization/linear_quant_modules.py b/torchao/quantization/linear_quant_modules.py
@@ -143,7 +143,7 @@ def __init__(
             "scales_and_zeros",
             torch.zeros(
                 (in_features // groupsize, out_features, 2),
-                dtype=self.scales_precision,
+                precision=self.scales_precision,
                 device=device,
             ),
         )

Original file line number	Diff line number	Diff line change
`@@ -143,7 +143,7 @@ def __init__(`
`143`	`143`	`"scales_and_zeros",`
`144`	`144`	`torch.zeros(`
`145`	`145`	`(in_features // groupsize, out_features, 2),`
`146`		`- dtype=self.scales_precision,`
	`146`	`+ precision=self.scales_precision,`
`147`	`147`	`device=device,`
`148`	`148`	`),`
`149`	`149`	`)`