Spaces:
Sleeping
Sleeping
Update requirements.txt
Browse files- requirements.txt +1 -1
requirements.txt
CHANGED
@@ -3,7 +3,7 @@ pytorch==2.1.2
|
|
3 |
torchvision==0.16.2
|
4 |
torchaudio==2.1.2
|
5 |
termcolor==2.4.0
|
6 |
-
pytorch-cuda
|
7 |
class_registry==2.1.2
|
8 |
datasets==3.1.0
|
9 |
griffe==0.48.0
|
|
|
3 |
torchvision==0.16.2
|
4 |
torchaudio==2.1.2
|
5 |
termcolor==2.4.0
|
6 |
+
pytorch-cuda==12.1
|
7 |
class_registry==2.1.2
|
8 |
datasets==3.1.0
|
9 |
griffe==0.48.0
|