requirements.txt

addict==2.4.0
albumentations==1.3.0
einops==0.6.0
git+https://github.com/huggingface/accelerate@78151f8
git+https://github.com/huggingface/diffusers@fa6d52d
gradio==3.20.0
imageio==2.25.0
imageio-ffmpeg==0.4.8
kornia==0.6.9
omegaconf==2.3.0
open-clip-torch==2.13.0
opencv-contrib-python==4.7.0.68
opencv-python-headless==4.7.0.68
prettytable==3.6.0
pytorch-lightning==1.9.0
safetensors==0.2.8
timm==0.6.12
torch==1.13.1
torchvision==0.14.1
transformers==4.26.1
xformers==0.0.16
yapf==0.32.0