From c26cc68e55e1811f91a3a610078d0a77e6e085f7 Mon Sep 17 00:00:00 2001 From: jhj0517 <97279763+jhj0517@users.noreply.github.com> Date: Wed, 12 Jun 2024 19:43:17 +0900 Subject: [PATCH 1/3] update requirements --- requirements.txt | 16 ++++++++++++---- 1 file changed, 12 insertions(+), 4 deletions(-) diff --git a/requirements.txt b/requirements.txt index 0a70c33..53d60e3 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,12 +1,14 @@ -torch==2.0.1 +--extra-index-url https://download.pytorch.org/whl/cu121 +torch==2.1.2 +torchvision==0.16.2 torchdiffeq==0.2.3 torchmetrics==1.2.1 torchsde==0.2.5 -torchvision==0.15.2 accelerate==0.29.3 av==11.0.0 clip @ https://github.com/openai/CLIP/archive/d50d76daa670286dd6cacf3bcd80b5e4823fc8e1.zip#sha256=b5842c25da441d6c581b53a5c60e0c2127ebafe0f746f8e15561a006c6c3be6a -decord==0.6.0 +decord==0.6.0; sys_platform != 'darwin' and platform_machine != 'arm64' +eva-decord==0.6.1; sys_platform == 'darwin' and platform_machine == 'arm64' diffusers>=0.24.0,<=0.27.2 einops==0.4.1 imageio==2.33.0 @@ -19,6 +21,12 @@ opencv-python==4.8.1.78 scikit-image==0.21.0 scikit-learn==1.3.2 transformers==4.33.1 -xformers==0.0.22 +xformers==0.0.23.post1 moviepy==1.0.3 wget==3.2 +wheel==0.43.0 +-f https://download.openmmlab.com/mmcv/dist/cu121/torch2.1/index.html +mmcv==2.1.0 +mmengine==0.10.4 +mmdet==3.3.0 +mmpose==1.3.1 \ No newline at end of file From 1a9e82657ba1e04399717431ded8ec3b977adfcf Mon Sep 17 00:00:00 2001 From: jhj0517 <97279763+jhj0517@users.noreply.github.com> Date: Wed, 12 Jun 2024 20:06:55 +0900 Subject: [PATCH 2/3] update readme --- README.md | 11 +---------- 1 file changed, 1 insertion(+), 10 deletions(-) diff --git a/README.md b/README.md index b9f7a25..0b8a31c 100644 --- a/README.md +++ b/README.md @@ -94,21 +94,12 @@ To prepare the Python environment and install additional packages such as opencv ### Build environment -We recommend a python version >=3.10 and cuda version =11.7. Then build environment as follows: +We recommend a python version >=3.10 and cuda version==12.1. Then build environment as follows: ```shell pip install -r requirements.txt ``` -### mmlab packages -```bash -pip install --no-cache-dir -U openmim -mim install mmengine -mim install "mmcv>=2.0.1" -mim install "mmdet>=3.1.0" -mim install "mmpose>=1.1.0" -``` - ### Download weights You can download weights manually as follows: From b2b5b3e6bdaa3d21895ed78e24cc8931b42a37c1 Mon Sep 17 00:00:00 2001 From: jhj0517 <97279763+jhj0517@users.noreply.github.com> Date: Wed, 7 Aug 2024 21:19:36 +0900 Subject: [PATCH 3/3] Remove `--extra-index-url` --- requirements.txt | 1 - 1 file changed, 1 deletion(-) diff --git a/requirements.txt b/requirements.txt index 53d60e3..eb5d6a8 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,4 +1,3 @@ ---extra-index-url https://download.pytorch.org/whl/cu121 torch==2.1.2 torchvision==0.16.2 torchdiffeq==0.2.3