musepose / requirements.txt
jhj0517
add triton
1121774
raw
history blame contribute delete
812 Bytes
--extra-index-url https://download.pytorch.org/whl/cu121
torch==2.1.2
torchvision==0.16.2
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
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
xformers==0.0.23.post1
gradio
triton