fix autoawq

huggingface · Sep 19, 2024 · f18330e · f18330e
1 parent 5c3a633
commit f18330e
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 1 deletion.
diff --git a/.github/workflows/images.yaml b/.github/workflows/images.yaml
@@ -69,3 +69,5 @@ jobs:
           labels: ${{ steps.meta.outputs.labels }}
           file: docker/${{ matrix.image_flavor }}/Dockerfile
           cache-from: type=registry,ref=${{ env.REGISTRY }}/${{ env.IMAGE_NAME }}:latest-${{ matrix.image_flavor }}
+          build-args: |
+            IMAGE_FLAVOR=${{ matrix.image_flavor }}
diff --git a/docker/rocm/Dockerfile b/docker/rocm/Dockerfile
@@ -47,6 +47,8 @@ fi
 # Install quantization libraries from source
 ENV ROCM_VERSION=5.7
 ENV PYTORCH_ROCM_ARCH="gfx900;gfx906;gfx908;gfx90a;gfx1030;gfx1100"
+ARG IMAGE_FLAVOR
+ENV IMAGE_FLAVOR=${IMAGE_FLAVOR}
 
 COPY scripts/install_quantization_libs.py /internal/install_quantization_libs.py
 RUN 

diff --git a/scripts/install_quantization_libs.py b/scripts/install_quantization_libs.py
@@ -37,7 +37,7 @@ def install_autoawq_from_source():
     print("Installing AutoAWQ and AutoAWQ_kernels packages.")
     autoawq_repo_path = os.path.join(EXTERNAL_REPOS_DIR, "AutoAWQ")
 
-    clone_or_pull_repo(f"https://github.com/casper-hansen/AutoAWQ", autoawq_repo_path)
+    clone_or_pull_repo("https://github.com/casper-hansen/AutoAWQ", autoawq_repo_path)
     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)