torch
numpy
einops
torchvision
tqdm
diffusers
accelerate
transformers
Pillow
poetry
git+https://github.com/openai/CLIP.git