use rocm docker

ModelCloud · Jan 8, 2025 · 50bf923 · 50bf923
1 parent 903c9f3
commit 50bf923
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/.github/workflows/unit_tests.yml b/.github/workflows/unit_tests.yml
@@ -126,11 +126,9 @@ jobs:
   rocm:
     runs-on: [self-hosted, rocm]
     container:
-      image: 10.0.14.248:5000/modelcloud/gptqmodel:github-ci-v6-base
+      image: 10.0.14.248:5000/modelcloud/gptqmodel:github-ci-v6-rocm
     steps:
       - uses: actions/checkout@v4
-#      - name: install rocm
-#        run: uv pip install "http://10.0.14.248/files/pytorch/torch-2.5.1+rocm6.2-cp312-cp312-linux_x86_64.whl" "http://10.0.14.248/files/pytorch/pytorch_triton_rocm-3.1.0-cp312-cp312-linux_x86_64.whl" -U --index-url https://download.pytorch.org/whl/rocm6.2
 
       - name: install
         run: pip install .