requirements.txt

decord==0.6.0
einops==0.3.0
imageio==2.9.0
numpy==1.24.2
omegaconf==2.1.1
opencv_python
pandas==2.0.0
Pillow==9.5.0
pytorch_lightning==2.3.1
PyYAML==6.0
setuptools==65.6.3
tqdm>=4.66.3
transformers==4.25.1
moviepy==1.0.3
av==12.2.0
timm==1.0.7
scikit-learn==1.5.0
open_clip_torch==2.22.0
kornia==0.7.3
albumentations==1.3.1
peft==0.11.1
accelerate==0.31.0
inflect==7.3.0
wandb==0.17.3
ipdb==0.13.13
huggingface-hub==0.23.4
gradio
spaces
-e git+https://github.com/tgxs002/HPSv2.git#egg=hpsv2