requirements.txt

omegaconf
torch
torchvision
einops
diffusers
transformers
pytorch_lightning==1.6.5
taming-transformers
kornia
git+https://github.com/openai/CLIP.git@main#egg=clip
git+https://github.com/davidbau/baukit.git