requirements.txt

-f https://download.pytorch.org/whl/torch_stable.html
git+https://github.com/openai/CLIP.git
torch==1.13.1+cu116
torchvision==0.14.1+cu116
pytorch-lightning==1.6.5
taming-transformers-rom1504==0.0.6
einops==0.6.0
kornia==0.6.11
transformers==4.27.4
dill==0.3.6
gradio==3.26.0
torchmetrics==0.11.4
omegaconf==2.3.0