Merge pull request #25 from kornia/chore/torch-link

chore (torch): update torch whl repo
kornia · Jul 29, 2024 · 9a5d39e · 9a5d39e
2 parents 6d1bb25 + 3c68307
commit 9a5d39e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/.github/actions/env/action.yml b/.github/actions/env/action.yml
@@ -53,9 +53,9 @@ runs:
       shell: bash
       run:   |
              if [ "$RUNNER_OS" == "Linux" ] || [ "$RUNNER_OS" == "Windows" ]; then
-                uv pip install --system torch==${{ inputs.pytorch-version }}+cpu ${{ inputs.extra-deps }} --find-links https://download.pytorch.org/whl/torch_stable.html
+                uv pip install --system torch==${{ inputs.pytorch-version }}+cpu ${{ inputs.extra-deps }} --find-links https://download.pytorch.org/whl/torch
              else
-                uv pip install --system torch==${{ inputs.pytorch-version }} ${{ inputs.extra-deps }} --find-links https://download.pytorch.org/whl/torch_stable.html
+                uv pip install --system torch==${{ inputs.pytorch-version }} ${{ inputs.extra-deps }} --find-links https://download.pytorch.org/whl/torch
              fi
 
     - if: ${{ contains(fromJson('["1.9.1", "1.10.2", "1.11.0", "1.12.1", "1.13.1", "2.0.1", "2.1.2", "2.2.2"]'), inputs.pytorch-version) }}