wip(tpu): training working, tgi not working

huggingface · Dec 6, 2024 · a121fb5 · a121fb5
1 parent 22aa768
commit a121fb5
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/.github/workflows/debug-dind.yml b/.github/workflows/debug-dind.yml
@@ -5,13 +5,17 @@ on:
 
 jobs:
   debug-dind:
+
     runs-on:
         group: gcp-ct5lp-hightpu-8t
     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 -e PJRT_DEVICE=TPU
 
     steps:
+      - name: Checkout code
+        uses: actions/checkout@v4
+
       - name: Install docker
         run: |
           apt-get update -y