fix torch
Browse files- requirements.txt +4 -4
requirements.txt
CHANGED
|
@@ -1,4 +1,4 @@
|
|
| 1 |
-
--extra-index-url https://download.pytorch.org/whl/
|
| 2 |
certifi==2025.1.31
|
| 3 |
charset-normalizer==3.4.1
|
| 4 |
cupy-cuda12x==13.3.0
|
|
@@ -38,9 +38,9 @@ safetensors==0.5.2
|
|
| 38 |
scipy==1.15.1
|
| 39 |
sympy==1.13.1
|
| 40 |
timm==1.0.14
|
| 41 |
-
torch==2.4.0
|
| 42 |
-
|
| 43 |
-
|
| 44 |
tqdm==4.67.1
|
| 45 |
triton==3.2.0
|
| 46 |
typing_extensions==4.12.2
|
|
|
|
| 1 |
+
--extra-index-url https://download.pytorch.org/whl/cu124
|
| 2 |
certifi==2025.1.31
|
| 3 |
charset-normalizer==3.4.1
|
| 4 |
cupy-cuda12x==13.3.0
|
|
|
|
| 38 |
scipy==1.15.1
|
| 39 |
sympy==1.13.1
|
| 40 |
timm==1.0.14
|
| 41 |
+
torch==2.4.0
|
| 42 |
+
torchvision==0.19.0
|
| 43 |
+
torchaudio==2.4.0
|
| 44 |
tqdm==4.67.1
|
| 45 |
triton==3.2.0
|
| 46 |
typing_extensions==4.12.2
|