fix autoawq

huggingface · Sep 19, 2024 · 1c71b40 · 1c71b40
1 parent f18330e
commit 1c71b40
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/scripts/install_quantization_libs.py b/scripts/install_quantization_libs.py
@@ -19,7 +19,7 @@ def clone_or_pull_repo(repo_url, repo_location_path):
 
 
 def process_setup_file_for_autoawq(setup_file_path):
-    print(f"Processing {setup_file_path} for AutoAWQ")
+    print(f"Processing setup.py for AutoAWQ")
 
     with open(setup_file_path, "r") as file:
         setup_content = file.read()
@@ -41,6 +41,9 @@ def install_autoawq_from_source():
     if os.environ.get("IMAGE_FLAVOR") in ["cuda", "rocm", "cuda-ort"]:
         autoawq_setup_file_path = os.path.join(autoawq_repo_path, "setup.py")
         process_setup_file_for_autoawq(autoawq_setup_file_path)
+        raise Exception("Success")
+
+    raise Exception("Failure")
     subprocess.run(
         f"cd {autoawq_repo_path} && INSTALL_KERNELS=1 {sys.executable} -m pip install .",
         shell=True,