upgrade pytorch to 2.5.1 in gptq tests

Signed-off-by: jiqing-feng <[email protected]>
huggingface · Dec 18, 2024 · 55bcdff · 55bcdff
1 parent 7125fe9
commit 55bcdff
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/.github/workflows/test_gptq.yml b/.github/workflows/test_gptq.yml
@@ -30,8 +30,7 @@ jobs:
       - name: Run tests
         uses: addnab/docker-run-action@v3
         with:
-          image: pytorch/pytorch:2.2.2-cuda12.1-cudnn8-runtime
-          # latest auto-gptq was built with pytorch 2.2 and cuda 12.1
+          image: pytorch/pytorch:2.5.1-cuda12.4-cudnn9-runtime
           options: |
             --rm
             --gpus all