numpy
torch
transformers
diffusers
accelerate
datasets
spacy
git+https://github.com/rom1504/clip-retrieval