Spaces:
Sleeping
Sleeping
Simon Duerr
commited on
Commit
•
d95db1d
1
Parent(s):
fa4efe3
change cuda version to cuda of image
Browse files- requirements.txt +2 -2
requirements.txt
CHANGED
|
@@ -1,4 +1,4 @@
|
|
| 1 |
-
torch==1.12.1+
|
| 2 |
transformers==4.29.1
|
| 3 |
einops
|
| 4 |
tqdm
|
|
@@ -11,4 +11,4 @@ matplotlib
|
|
| 11 |
seaborn
|
| 12 |
black
|
| 13 |
ipython
|
| 14 |
-
--extra-index-url https://download.pytorch.org/whl/
|
|
|
|
| 1 |
+
torch==1.12.1+cu113
|
| 2 |
transformers==4.29.1
|
| 3 |
einops
|
| 4 |
tqdm
|
|
|
|
| 11 |
seaborn
|
| 12 |
black
|
| 13 |
ipython
|
| 14 |
+
--extra-index-url https://download.pytorch.org/whl/cu113
|