ikuinen99 commited on
Commit
d122b2d
1 Parent(s): 411cfb5
Files changed (2) hide show
  1. pre-requirements.txt +3 -2
  2. requirements.txt +2 -1
pre-requirements.txt CHANGED
@@ -1,5 +1,6 @@
1
- torch
2
- torchvision
 
3
  torchaudio
4
  accelerate
5
  aiohttp==3.8.4
 
1
+ --extra-index-url https://download.pytorch.org/whl/cu113
2
+ torch==1.12.1
3
+ torchvision==0.13.1
4
  torchaudio
5
  accelerate
6
  aiohttp==3.8.4
requirements.txt CHANGED
@@ -1,4 +1,5 @@
1
  git+https://github.com/facebookresearch/segment-anything.git
2
  git+https://github.com/IDEA-Research/GroundingDINO.git
3
  fairscale
4
- mmcv==2.0.0 --extra-index-url https://download.openmmlab.com/mmcv/dist/cu117/torch2.0/index.html
 
 
1
  git+https://github.com/facebookresearch/segment-anything.git
2
  git+https://github.com/IDEA-Research/GroundingDINO.git
3
  fairscale
4
+ --extra-index-url https://download.openmmlab.com/mmcv/dist/cu113/torch1.12/index.html
5
+ mmcv==2.0.0rc4