fix typo

pytorch · Apr 9, 2024 · 6aceb29 · 6aceb29
1 parent 42ee62f
commit 6aceb29
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/quantize.py b/quantize.py
@@ -66,7 +66,7 @@ def quantize_model(model: nn.Module, quantize_options):
             ).quantized_model()
         elif quantizer == "linear:hqq":
             linears_quantized = True
-            model = WeightOnlyInt4HqqQuantHandler,
+            model = WeightOnlyInt4HqqQuantHandler(
                 model,
                 **q_kwargs
             ).quantized_model()