musepose / requirements.txt
jhj0517
update requirements.txt
403285a
raw
history blame
922 Bytes
torch==2.1.2 --install-option='--index-url https://download.pytorch.org/whl/cu121'
torchvision==0.16.2 --install-option='--index-url https://download.pytorch.org/whl/cu121'
av==11.0.0
torchdiffeq==0.2.3
torchmetrics==1.2.1
torchsde==0.2.5
accelerate==0.29.3
clip @ https://github.com/openai/CLIP/archive/d50d76daa670286dd6cacf3bcd80b5e4823fc8e1.zip#sha256=b5842c25da441d6c581b53a5c60e0c2127ebafe0f746f8e15561a006c6c3be6a
decord==0.6.0
diffusers>=0.24.0,<=0.27.2
einops==0.4.1
imageio==2.33.0
imageio-ffmpeg==0.4.9
ffmpeg-python==0.2.0
omegaconf==2.2.3
open-clip-torch==2.20.0
opencv-contrib-python==4.8.1.78
opencv-python==4.8.1.78
scikit-image==0.21.0
scikit-learn==1.3.2
transformers==4.33.1
xformers==0.0.22
moviepy==1.0.3
wget==3.2
wheel==0.43.0
mmcv==2.1.0 --install-option='--find-links https://download.openmmlab.com/mmcv/dist/cu121/torch2.1/index.html'
mmengine==0.10.4
mmdet==3.3.0
mmpose==1.3.1
xformers==0.0.22