jerpelhan commited on
Commit
3e76066
·
verified ·
1 Parent(s): f7ef64f

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +3 -3
requirements.txt CHANGED
@@ -1,8 +1,8 @@
1
  --extra-index-url https://download.pytorch.org/whl/cu121
2
 
3
- torch==2.1.0+cu121
4
- torchvision==0.16.0+cu121
5
- torchaudio==2.1.0+cu121
6
  multiscaledeformableattention @ https://huggingface.co/datasets/jerpelhan/geco2-assets/resolve/main/wheels/multiscaledeformableattention-1.0-cp310-cp310-linux_x86_64.whl
7
 
8
  huggingface-hub
 
1
  --extra-index-url https://download.pytorch.org/whl/cu121
2
 
3
+ torch==2.1.0
4
+ torchvision==0.16.0
5
+ torchaudio==2.1.0
6
  multiscaledeformableattention @ https://huggingface.co/datasets/jerpelhan/geco2-assets/resolve/main/wheels/multiscaledeformableattention-1.0-cp310-cp310-linux_x86_64.whl
7
 
8
  huggingface-hub