pablodawson commited on
Commit
5cccf5d
·
1 Parent(s): 5c44dfb

Update requirements.txt

Browse files
Files changed (1) hide show
  1. 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.9.0+cu111
6
  transformers
7
  pytorch_lightning
8
- -f https://shi-labs.com/natten/wheels/cu111/torch1.9/index.html
9
- natten==0.14.2
 
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