jhj0517 commited on
Commit
3dd6f82
1 Parent(s): 403285a

update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +5 -5
requirements.txt CHANGED
@@ -1,5 +1,6 @@
1
- torch==2.1.2 --install-option='--index-url https://download.pytorch.org/whl/cu121'
2
- torchvision==0.16.2 --install-option='--index-url https://download.pytorch.org/whl/cu121'
 
3
  av==11.0.0
4
  torchdiffeq==0.2.3
5
  torchmetrics==1.2.1
@@ -19,12 +20,11 @@ opencv-python==4.8.1.78
19
  scikit-image==0.21.0
20
  scikit-learn==1.3.2
21
  transformers==4.33.1
22
- xformers==0.0.22
23
  moviepy==1.0.3
24
  wget==3.2
25
  wheel==0.43.0
26
- mmcv==2.1.0 --install-option='--find-links https://download.openmmlab.com/mmcv/dist/cu121/torch2.1/index.html'
27
  mmengine==0.10.4
28
  mmdet==3.3.0
29
  mmpose==1.3.1
30
- xformers==0.0.22
 
1
+ --extra-index-url https://download.pytorch.org/whl/cu121
2
+ torch==2.1.2
3
+ torchvision==0.16.2
4
  av==11.0.0
5
  torchdiffeq==0.2.3
6
  torchmetrics==1.2.1
 
20
  scikit-image==0.21.0
21
  scikit-learn==1.3.2
22
  transformers==4.33.1
 
23
  moviepy==1.0.3
24
  wget==3.2
25
  wheel==0.43.0
26
+ mmcv==2.1.0 --find-links https://download.openmmlab.com/mmcv/dist/cu121/torch2.1/index.html
27
  mmengine==0.10.4
28
  mmdet==3.3.0
29
  mmpose==1.3.1
30
+ xformers==0.0.23.post1