pablodawson
commited on
Commit
·
5cccf5d
1
Parent(s):
5c44dfb
Update requirements.txt
Browse files- requirements.txt +4 -4
requirements.txt
CHANGED
@@ -1,9 +1,9 @@
|
|
1 |
huggingface_hub
|
2 |
setuptools_rust
|
3 |
safetensors>=0.3
|
4 |
-
--extra-index-url https://download.pytorch.org/whl
|
5 |
-
torch==1.
|
6 |
transformers
|
7 |
pytorch_lightning
|
8 |
-
-f https://shi-labs.com/natten/wheels/
|
9 |
-
natten==0.14.
|
|
|
1 |
huggingface_hub
|
2 |
setuptools_rust
|
3 |
safetensors>=0.3
|
4 |
+
--extra-index-url https://download.pytorch.org/whl/cu116
|
5 |
+
torch==1.13.1+cu116
|
6 |
transformers
|
7 |
pytorch_lightning
|
8 |
+
-f https://shi-labs.com/natten/wheels/cu116/torch1.13/index.html
|
9 |
+
natten==0.14.5
|