diff --git a/.github/workflows/test_tpu_tgi.yml b/.github/workflows/test-pytorch-xla-tpu-tgi.yml similarity index 92% rename from .github/workflows/test_tpu_tgi.yml rename to .github/workflows/test-pytorch-xla-tpu-tgi.yml index ab06082b..f089bf35 100644 --- a/.github/workflows/test_tpu_tgi.yml +++ b/.github/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"