update docker image!

huggingface · Oct 11, 2024 · 9b067da · 9b067da
1 parent ce0b61f
commit 9b067da
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/.github/workflows/benchmark.yml b/.github/workflows/benchmark.yml
@@ -19,7 +19,7 @@ jobs:
     runs-on:
       group: aws-g5-4xlarge-cache
     container:
-      image: huggingface/transformers-all-latest-gpu
+      image: huggingface/transformers-pytorch-gpu
       options: --gpus all --privileged --ipc host
     steps:
       - name: Get repo