admin commited on
Commit ·
7e28019
1
Parent(s): 3d3fab6
try avoid torch 2.6
Browse files- requirements.txt +2 -2
requirements.txt
CHANGED
|
@@ -1,6 +1,6 @@
|
|
| 1 |
-
torch==2.
|
| 2 |
-f https://download.pytorch.org/whl/torch
|
| 3 |
-
torchvision==0.
|
| 4 |
-f https://download.pytorch.org/whl/torchvision
|
| 5 |
scipy
|
| 6 |
omegaconf
|
|
|
|
| 1 |
+
torch==2.5.1+cu118
|
| 2 |
-f https://download.pytorch.org/whl/torch
|
| 3 |
+
torchvision==0.20.1+cu118
|
| 4 |
-f https://download.pytorch.org/whl/torchvision
|
| 5 |
scipy
|
| 6 |
omegaconf
|