udpate cli argument

huggingface · Aug 15, 2024 · f3058f1 · f3058f1
1 parent 1353433
commit f3058f1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/optimum_benchmark/backends/pytorch/backend.py b/optimum_benchmark/backends/pytorch/backend.py
@@ -293,7 +293,7 @@ def process_quantization_config(self) -> None:
                 raise ImportError(
                     "GPTQ quantization requires the AutoGPTQ package. "
                     "Please install it from source at `https://github.com/AutoGPTQ/AutoGPTQ`"
-                    "Or run optimum-benchmark +install_auto_gptq=True"
+                    "Or run optimum-benchmark +install_auto_gptq_from_source=True"
                 )
 
             self.quantization_config = GPTQConfig(
@@ -308,7 +308,7 @@ def process_quantization_config(self) -> None:
                 raise ImportError(
                     "AWQ quantization requires the AutoAWQ package. "
                     "Please install it from source at `https://github.com/casper-hansen/AutoAWQ`"
-                    "Or run optimum-benchmark +install_auto_awq=True"
+                    "Or run optimum-benchmark +install_auto_awq_from_source=True"
                 )
 
             self.quantization_config = AwqConfig(