jhj0517 commited on
Commit ·
3dd6f82
1
Parent(s): 403285a
update requirements.txt
Browse files- requirements.txt +5 -5
requirements.txt
CHANGED
|
@@ -1,5 +1,6 @@
|
|
| 1 |
-
|
| 2 |
-
|
|
|
|
| 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 --
|
| 27 |
mmengine==0.10.4
|
| 28 |
mmdet==3.3.0
|
| 29 |
mmpose==1.3.1
|
| 30 |
-
xformers==0.0.
|
|
|
|
| 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
|