fix workflow

huggingface · Nov 27, 2024 · 726a54e · 726a54e
1 parent 5556ebb
commit 726a54e
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/.github/workflows/test-pytorch-xla-tpu-tgi-integration.yml b/.github/workflows/test-pytorch-xla-tpu-tgi-integration.yml
@@ -19,9 +19,9 @@ jobs:
       group: gcp-ct5lp-hightpu-8t
 
     # the error (http://metadata.google.internal/computeMetadata/v1 not accesible ) is the 
-    container:
-        image: us-central1-docker.pkg.dev/tpu-pytorch-releases/docker/xla:r2.4.0_3.10_tpuvm
-        options: --shm-size "16gb" --ipc host --privileged ${{ vars.V5_LITEPOD_8_ENV}} -v /mnt/hf_cache:/mnt/hf_cache
+    # container:
+    #     image: us-central1-docker.pkg.dev/tpu-pytorch-releases/docker/xla:r2.4.0_3.10_tpuvm
+    #     options: --shm-size "16gb" --ipc host --privileged ${{ vars.V5_LITEPOD_8_ENV}} -v /mnt/hf_cache:/mnt/hf_cache
 
     env:
       PJRT_DEVICE: TPU