torch
safetensors
pytorch-lightning
git+https://github.com/openai/CLIP.git
numpy
pillow