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