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

huggingface · Apr 11, 2024 · 540efc6 · 540efc6
1 parent 0792bd9
commit 540efc6
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/.github/workflows/test-pytorch-xla-tpu-tgi.yml b/.github/workflows/test-pytorch-xla-tpu-tgi.yml
@@ -30,11 +30,11 @@ jobs:
     steps:        
       - name: Checkout
         uses: actions/checkout@v4
-      - name: test
-        run:  sleep 60m
+      #- name: test
+      #  run:  sleep 60m
 
-      #- 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: |