Spaces:
Runtime error
Runtime error
root
commited on
Commit
•
8dbe2ab
1
Parent(s):
fe1b81a
req
Browse files- requirements.txt +3 -2
requirements.txt
CHANGED
@@ -1,5 +1,5 @@
|
|
1 |
-
|
2 |
-
torch==1.11.0
|
3 |
torchvision
|
4 |
numpy
|
5 |
transformers==4.19.2
|
@@ -10,3 +10,4 @@ regex==2022.9.13
|
|
10 |
timm==0.6.12
|
11 |
imageio
|
12 |
gradio
|
|
|
|
1 |
+
--find-links https://download.pytorch.org/whl/torch_stable.html
|
2 |
+
torch==1.11.0+cu113
|
3 |
torchvision
|
4 |
numpy
|
5 |
transformers==4.19.2
|
|
|
10 |
timm==0.6.12
|
11 |
imageio
|
12 |
gradio
|
13 |
+
# cudatoolkit==11.3.1
|