diff --git a/.github/actions/env/action.yml b/.github/actions/env/action.yml index f75f130..a708e11 100644 --- a/.github/actions/env/action.yml +++ b/.github/actions/env/action.yml @@ -51,7 +51,7 @@ runs: - if: ${{ contains(fromJson('["nightly"]'), inputs.pytorch-version ) == false}} name: Install pytorch shell: bash - run: uv pip install --system torch==${{ inputs.pytorch-version }}+cpu ${{ inputs.extra-deps }} --find-links https://download.pytorch.org/whl/torch_stable.html + run: uv pip install --system torch==${{ inputs.pytorch-version }}+cpu ${{ inputs.extra-deps }} --extra-index-url https://download.pytorch.org/whl/cpu - if: ${{ contains(fromJson('["1.9.1"]'), inputs.pytorch-version) }} name: Install accelerate for old torchs