Sirus1 commited on
Commit
1c88419
·
1 Parent(s): 30995bf

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +6 -3
requirements.txt CHANGED
@@ -1,6 +1,9 @@
1
- torch
2
- torchvision
3
- torchaudio
 
 
 
4
  transformers
5
  timm==0.4.12
6
  openai
 
1
+ --extra-index-url https://download.pytorch.org/whl
2
+ torch==1.9.0+cu111
3
+ torchvision==0.10.0+cu111
4
+ torchaudio==0.9.0
5
+ -f https://dl.fbaipublicfiles.com/detectron2/wheels/cu111/torch1.9/index.html
6
+ detectron2==0.6
7
  transformers
8
  timm==0.4.12
9
  openai