add quanization lib to an install script

Makefile

@@ -109,8 +109,7 @@ install_cli_cpu_neural_compressor:
 
 install_cli_cuda_pytorch:
 	pip install -e .[testing,timm,diffusers,peft,autoawq,auto-gptq,bitsandbytes,deepspeed]
-	optimum-benchmark +install_auto_awq_from_source=True
-	optimum-benchmark +install_auto_gptq_from_source=True
+	curl -s https://raw.githubusercontent.com/huggingface/optimum-benchmark/main/scripts/total_tests_runs.py | python - --install-autoawq --install-auto-gptq
 
 install_cli_rocm_pytorch:
 	pip install -e .[testing,timm,diffusers,peft,autoawq,auto-gptq,deepspeed]

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_from_source=True"
+                    "Or `curl -s https://raw.githubusercontent.com/huggingface/optimum-benchmark/main/scripts/total_tests_runs.py | python - --install-auto-gptq`"
                 )
 
             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_from_source=True"
+                    "Or `curl -s https://raw.githubusercontent.com/huggingface/optimum-benchmark/main/scripts/total_tests_runs.py | python - --install-autoawq`"
                 )
 
             self.quantization_config = AwqConfig(

optimum_benchmark/cli.py

@@ -6,8 +6,6 @@
 from hydra.core.config_store import ConfigStore
 from omegaconf import DictConfig, OmegaConf
 
-from optimum_benchmark.install_utils import InstallConfig, install_autoawq, install_autogptq
-
 from . import (
     Benchmark,
     BenchmarkConfig,
@@ -57,20 +55,10 @@
 cs.store(group="launcher", name=ProcessConfig.name, node=ProcessConfig)
 cs.store(group="launcher", name=TorchrunConfig.name, node=TorchrunConfig)
 
-cs.store(name="install_config", node=InstallConfig)
-
 
 # optimum-benchmark
 @hydra.main(version_base=None)
 def main(config: DictConfig) -> None:
-    if "install_auto_awq_from_source" in config and config.install_auto_awq_from_source:
-        install_autoawq()
-        return
-
-    if "install_auto_gptq_from_source" in config and config.install_auto_gptq_from_source:
-        install_autogptq()
-        return
-
     log_level = os.environ.get("LOG_LEVEL", "INFO")
     log_to_file = os.environ.get("LOG_TO_FILE", "1") == "1"
     override_benchmarks = os.environ.get("OVERRIDE_BENCHMARKS", "0") == "1"