Fixing import typo in quantize.py

Jack-Khuu · metascroy · commit f9a7bb9a6285 · 2025-01-29T17:15:26.000-08:00
diff --git a/torchchat/utils/quantize.py b/torchchat/utils/quantize.py
@@ -967,8 +967,7 @@ def quantized_model(self) -> nn.Module:
     torchao_experimental_quant_api_spec.loader.exec_module(
         torchao_experimental_quant_api
     )
-    from torchao_experimental_quant_api import (
-        from torchao_experimental_quant_api import UIntxWeightOnlyLinearQuantizer
+    from torchao_experimental_quant_api import UIntxWeightOnlyLinearQuantizer
     quantizer_class_dict["linear:afpwx"] = UIntxWeightOnlyLinearQuantizer
 
     # Try loading custom op

Original file line number	Diff line number	Diff line change
`@@ -967,8 +967,7 @@ def quantized_model(self) -> nn.Module:`
`967`	`967`	`torchao_experimental_quant_api_spec.loader.exec_module(`
`968`	`968`	`torchao_experimental_quant_api`
`969`	`969`	`)`
`970`		`- from torchao_experimental_quant_api import (`
`971`		`- from torchao_experimental_quant_api import UIntxWeightOnlyLinearQuantizer`
	`970`	`+ from torchao_experimental_quant_api import UIntxWeightOnlyLinearQuantizer`
`972`	`971`	`quantizer_class_dict["linear:afpwx"] = UIntxWeightOnlyLinearQuantizer`
`973`	`972`
`974`	`973`	`# Try loading custom op`