fix typo

pytorch · Apr 10, 2024 · 0a6383a · 0a6383a
1 parent c5fac59
commit 0a6383a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/quantize.py b/quantize.py
@@ -54,7 +54,7 @@ def quantize_model(model: nn.Module, quantize_options):
             ).quantized_model()
         elif quantizer == "linear:int4":
             linears_quantized = True
-            model = Int8DynActInt4WeightQuantHandler(
+            model = WeightOnlyInt4QuantHandler(
                 model,
                 **q_kwargs
             ).quantized_model()