Update fork-maintenance.yml

Cemberk · Sep 3, 2024 · 61d05b5 · 61d05b5
1 parent 0bbde42
commit 61d05b5
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/.github/workflows/fork-maintenance.yml b/.github/workflows/fork-maintenance.yml
@@ -297,13 +297,12 @@ jobs:
                   - tifffile==2024.7.24
                   - timeout-decorator==0.5.0
                   - timm==1.0.9
-                  - "pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/rocm6.1"
                   - tokenizers==0.19.1
                   - toml==0.10.2
                   - tomli==2.0.1
-                  - torch==2.4.0+https://download.pytorch.org/whl/rocm6.1
-                  - torchaudio==2.4.0+https://download.pytorch.org/whl/rocm6.1
-                  - torchvision==0.19.0+https://download.pytorch.org/whl/rocm6.1
+                  - torch==2.4.0 --extra-index-url https://download.pytorch.org/whl/rocm6.1
+                  - torchaudio==2.4.0 --extra-index-url https://download.pytorch.org/whl/rocm6.1
+                  - torchvision==0.19.0 --extra-index-url https://download.pytorch.org/whl/rocm6.1
                   - tornado==6.4.1
                   - tqdm==4.66.4
                   - transformers==4.43.0.dev0