modify attribute

huggingface · Jul 26, 2023 · 1df19a1 · 1df19a1
1 parent 3633d43
commit 1df19a1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/optimum/gptq/quantizer.py b/optimum/gptq/quantizer.py
@@ -447,7 +447,7 @@ def tmp(_, input, output):
         if not self.pack_sequentially:
             self.pack_model(model=model, quantizers=quantizers)
 
-        model._is_quantized_gptq = True
+        model._is_gptq_quantized = True
         if has_config:
             model.config.use_cache = use_cache
             model.config.quantization_config = self.to_dict()