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

huggingface · Apr 17, 2024 · 0e869d5 · 0e869d5
1 parent 0523558
commit 0e869d5
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/.github/workflows/test-pytorch-xla-tpu-tgi.yml b/.github/workflows/test-pytorch-xla-tpu-tgi.yml
@@ -29,6 +29,8 @@ jobs:
         #uses: actions/checkout@v4
       - name: slepp
         run: |
+          apt update
+          apt install pip 
           pip install 'jax[tpu]' -f https://storage.googleapis.com/jax-releases/libtpu_releases.html
           pip install --upgrade pip
           python -c 'import jax; print("TPU cores:", jax.device_count())'