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