requirements.txt

absl-py==1.4.0
accelerate==0.12.0
einops==0.6.0
ftfy==6.1.1
git+https://github.com/openai/CLIP.git@a9b1bf5
gradio==3.21.0
huggingface-hub==0.13.2
ml-collections==0.1.1
torch==1.13.1
torchvision==0.14.1
transformers==4.23.1
triton==2.0.0
xformers==0.0.16