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

huggingface · Apr 10, 2024 · fd0e279 · fd0e279
1 parent aae1d91
commit fd0e279
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/.github/workflows/test-pytorch-xla-tpu-tgi.yml b/.github/workflows/test-pytorch-xla-tpu-tgi.yml
@@ -31,8 +31,7 @@ jobs:
         uses: actions/checkout@v4
 
       - name: Checking Pytorch/XLA installation
-        run: sleep 60m
-        #run: python -c "import torch_xla.core.xla_model as xm; assert xm.xla_device().type == 'xla', 'XLA device not available'"
+        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: |