WIP

huggingface · Feb 28, 2024 · afcbb62 · afcbb62
1 parent c99ed3e
commit afcbb62
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/.github/workflows/test_tpu_tgi.yml → ...ub/workflows/test-pytorch-xla-tpu-tgi.yml b/.github/workflows/test_tpu_tgi.yml → ...ub/workflows/test-pytorch-xla-tpu-tgi.yml
@@ -21,12 +21,12 @@ jobs:
     runs-on: optimum-tpu
     container:
       image: us-central1-docker.pkg.dev/tpu-pytorch-releases/docker/xla:r2.2.0_3.10_tpuvm
-      options: --shm-size "16gb" --ipc host -v /mnt/hf_cache:/mnt/cache/ --privileged
+      options: --shm-size "16gb" --ipc host --privileged
     env:
       PJRT_DEVICE: TPU
     steps:
       - name: Checkout
-        uses: actions/checkout@v2
+        uses: actions/checkout@v4
       - name: Create venv and install Pytorch/XLA
         run: |
           echo "PJRT_DEVICE set to $PJRT_DEVICE"