Make-A-Protagonist-inference / requirements.txt
HeliosZhao
update dockerfile
2a9ba02
accelerate==0.18.0
bitsandbytes==0.35.4
decord==0.6.0
diffusers[torch]==0.15.0
einops==0.6.0
ftfy==6.1.1
gradio
huggingface-hub
imageio==2.25.0
imageio-ffmpeg==0.4.8
omegaconf
Pillow
python-slugify
tensorboard
# torch==1.13.1
# torchvision==0.14.1
transformers==4.27.4
# triton
# xformers==0.0.17
ipdb
opencv-python
addict
yapf
onnxruntime
pycocotools
PyYAML
requests
setuptools
supervision
termcolor
timm
nltk
fairscale