Update test-pytorch-xla-tpu-tgi.yml

huggingface · Apr 10, 2024 · 25fde9b · 25fde9b
1 parent 75c1234
commit 25fde9b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/.github/workflows/test-pytorch-xla-tpu-tgi.yml b/.github/workflows/test-pytorch-xla-tpu-tgi.yml
@@ -22,8 +22,8 @@ jobs:
     runs-on: tpu-runners-dind2
     container:
       # Use an image that works with TPU with Pytorch 2.3.0 (release was not working)
-      #image: us-central1-docker.pkg.dev/tpu-pytorch-releases/docker/xla@sha256:8f1dcd5b03f993e4da5c20d17c77aff6a5f22d5455f8eb042d2e4b16ac460526
-      image: us-central1-docker.pkg.dev/tpu-pytorch-releases/docker/xla@sha256:a0b5294d9e00d01876105e5afc5a187ea4c3e481f6f5a407b6621f9f1af7d9b2
+      image: us-central1-docker.pkg.dev/tpu-pytorch-releases/docker/xla@sha256:8f1dcd5b03f993e4da5c20d17c77aff6a5f22d5455f8eb042d2e4b16ac460526
+      #image: us-central1-docker.pkg.dev/tpu-pytorch-releases/docker/xla@sha256:a0b5294d9e00d01876105e5afc5a187ea4c3e481f6f5a407b6621f9f1af7d9b2
       options: --shm-size "16gb" --ipc host --privileged
     env:
       PJRT_DEVICE: TPU