correct attribute

huggingface · Jul 4, 2024 · cd1ea69 · cd1ea69
1 parent 04a334a
commit cd1ea69
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/scripts/quantize.py b/scripts/quantize.py
@@ -152,7 +152,7 @@ def quantize_q8(
     )
 
     quantizer.quantize_model()
-    check_and_save_model(quantizer.model, save_path)
+    check_and_save_model(quantizer.model.model, save_path)
 
 
 def quantize_fp16(