diff --git a/.github/workflows/test-pytorch-xla-tpu-tgi.yml b/.github/workflows/test-pytorch-xla-tpu-tgi.yml index 295e18e3..53da56b3 100644 --- a/.github/workflows/test-pytorch-xla-tpu-tgi.yml +++ b/.github/workflows/test-pytorch-xla-tpu-tgi.yml @@ -27,13 +27,9 @@ jobs: steps: - name: Checkout uses: actions/checkout@v4 - - name: test CI - run: | - pip install --upgrade pip - 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: 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: |