jerpelhan commited on
Commit
a8a8a71
·
verified ·
1 Parent(s): 5d20ea4

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +2 -3
requirements.txt CHANGED
@@ -1,11 +1,10 @@
1
- --extra-index-url https://download.pytorch.org/whl/cu126
 
2
 
3
  torch==2.11.0
4
  torchvision==0.26.0
5
  torchaudio==2.11.0
6
 
7
- nvidia-cuda-runtime-cu12
8
-
9
  multiscaledeformableattention @ https://huggingface.co/datasets/jerpelhan/geco2-assets/resolve/main/wheels/MultiScaleDeformableAttention-1.0-cp310-cp310-linux_x86_64.whl
10
 
11
  huggingface-hub
 
1
+ --index-url https://download.pytorch.org/whl/cu128
2
+ --extra-index-url https://pypi.org/simple
3
 
4
  torch==2.11.0
5
  torchvision==0.26.0
6
  torchaudio==2.11.0
7
 
 
 
8
  multiscaledeformableattention @ https://huggingface.co/datasets/jerpelhan/geco2-assets/resolve/main/wheels/MultiScaleDeformableAttention-1.0-cp310-cp310-linux_x86_64.whl
9
 
10
  huggingface-hub