Skip to content

Commit

Permalink
Update test-pytorch-xla-tpu-tgi.yml
Browse files Browse the repository at this point in the history
  • Loading branch information
paulinebm authored Apr 10, 2024
1 parent 92a6f82 commit 02e0071
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion .github/workflows/test-pytorch-xla-tpu-tgi.yml
Original file line number Diff line number Diff line change
Expand Up @@ -31,7 +31,8 @@ jobs:
uses: actions/checkout@v4

- 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'"
run: sleep 60m
#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: |
Expand Down

0 comments on commit 02e0071

Please sign in to comment.