Spaces:
Sleeping
Sleeping
Update requirements.txt
Browse files- requirements.txt +2 -1
requirements.txt
CHANGED
@@ -5,4 +5,5 @@ scipy
|
|
5 |
accelerate
|
6 |
torchvision==0.15.2
|
7 |
matplotlib
|
8 |
-
|
|
|
|
5 |
accelerate
|
6 |
torchvision==0.15.2
|
7 |
matplotlib
|
8 |
+
torch==2.4.1
|
9 |
+
--extra-index-url https://download.pytorch.org/whl/cu118
|