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 15, 2024
1 parent 8edc62b commit 8e0b2b3
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 @@ -27,7 +27,8 @@ jobs:
steps:
- name: Checkout
uses: actions/checkout@v4

- name: Install dependencies
run: pip install torch~=2.2.0 torch_xla[tpu]~=2.2.0 -f https://storage.googleapis.com/libtpu-releases/index.html
- 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'"

Expand Down

0 comments on commit 8e0b2b3

Please sign in to comment.