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

huggingface · Apr 15, 2024 · 3f6b853 · 3f6b853
1 parent 880c18d
commit 3f6b853
Showing 1 changed file with 9 additions and 5 deletions.
diff --git a/.github/workflows/test-pytorch-xla-tpu-tgi.yml b/.github/workflows/test-pytorch-xla-tpu-tgi.yml
@@ -27,10 +27,14 @@ jobs:
     steps:
       - name: Checkout
         uses: actions/checkout@v4
+      - name: test CI
+        run: |
+          pip install --upgrade pip
+          python -c "import torch_xla.core.xla_model as xm; assert xm.xla_device().type == 'xla', 'XLA device not available'"
 
-      - name: Checking Pytorch/XLA installation
-        run: python -c "import torch_xla.core.xla_model as xm; assert xm.xla_device().type == 'xla', 'XLA device not available'"
+      #- name: Checking Pytorch/XLA installation
+      #  run: python -c "import torch_xla.core.xla_model as xm; assert xm.xla_device().type == 'xla', 'XLA device not available'"
 
-      - name: Build and test TGI server
-        run: |
-          HF_TOKEN=${{ secrets.HF_TOKEN_OPTIMUM_TPU_CI }} make tgi_test
+      #- name: Build and test TGI server
+      #  run: |
+      #    HF_TOKEN=${{ secrets.HF_TOKEN_OPTIMUM_TPU_CI }} make tgi_test