From 3f6b8534a19adec73fdc2109d19363a310d4b94a Mon Sep 17 00:00:00 2001
From: Pauline Bailly-Masson <155966238+paulinebm@users.noreply.github.com>
Date: Mon, 15 Apr 2024 15:24:08 +0200
Subject: [PATCH] Update test-pytorch-xla-tpu-tgi.yml

---
 .github/workflows/test-pytorch-xla-tpu-tgi.yml | 14 +++++++++-----
 1 file changed, 9 insertions(+), 5 deletions(-)

diff --git a/.github/workflows/test-pytorch-xla-tpu-tgi.yml b/.github/workflows/test-pytorch-xla-tpu-tgi.yml
index 35caad48..295e18e3 100644
--- 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