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 8e0b2b3 commit 78a9366
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions .github/workflows/test-pytorch-xla-tpu-tgi.yml
Original file line number Diff line number Diff line change
Expand Up @@ -27,8 +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: 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 78a9366

Please sign in to comment.