requirements.txt

torch==1.13.1
torchvision==0.14.1
xformers==0.0.16
omegaconf==2.1.1
albumentations==1.3.0
opencv-python
imageio==2.9.0
imageio-ffmpeg==0.4.2
pytorch-lightning==1.4.2
test-tube>=0.7.5
streamlit==1.12.1
einops==0.3.0
git+https://github.com/openai/CLIP.git
protobuf~=3.20.1
torchmetrics==0.6.0
transformers==4.19.2
kornia==0.6.0
gradio==3.16.0