diffusers==0.33.1
gradio==4.44.1
torch>=2.4.0
numpy
matplotlib
pillow
wandb
einops
transformers
accelerate
huggingface_hub
git+https://github.com/wendlerc/clip-retrieval.git
sentencepiece