fix

huggingface · Jan 29, 2025 · 8fa9d71 · 8fa9d71
1 parent 4cf7b38
commit 8fa9d71
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/transformers/quantizers/auto.py b/src/transformers/quantizers/auto.py
@@ -217,4 +217,5 @@ def supports_quant_method(quantization_config_dict):
                 f" {list(AUTO_QUANTIZER_MAPPING.keys())}. Hence, we will skip the quantization. "
                 "To remove the warning, you can delete the quantization_config attribute in config.json"
             )
-        return False
+            return False
+        return True