requirements.txt

pydantic
tqdm
click
easydict
tabulate
imageio
einops
matplotlib
omegaconf==2.1.1
torchmetrics==0.6.0
pytorch-lightning==1.4.2
transformers
kornia==0.6.0
git+https:///github.com/openai/CLIP.git#egg=clip
imageio[ffmpeg]
imageio[pyav]
taming-transformers