Fixing import typo in quantize.py

pytorch · Jan 30, 2025 · f9a7bb9 · f9a7bb9
1 parent 525701d
commit f9a7bb9
Showing 1 changed file with 1 addition and 2 deletions.
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