debug

huggingface · Feb 27, 2024 · 9b7ca11 · 9b7ca11
1 parent affa9b0
commit 9b7ca11
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/.github/workflows/test_tpu_tgi.yml b/.github/workflows/test_tpu_tgi.yml
@@ -26,6 +26,7 @@ jobs:
         uses: actions/checkout@v2
       - name: Create venv and install Pytorch/XLA
         run: |
+          echo "PJRT_DEVICE set to $PJRT_DEVICE"
           echo "Creating virtual environment..."
           python3 -m venv venv-tpu-pytorch
           source venv-tpu-pytorch/bin/activate