requirements.txt

accelerate==1.1.1
deepspeed==0.14.4
einops==0.8.0
transformers==4.43.3
huggingface-hub
diffusers==0.30.1
sentencepiece==0.2.0
gradio==5.22.0
opencv-python
matplotlib
safetensors==0.4.5
scipy==1.10.1
numpy==1.24.4
onnxruntime-gpu
# httpx==0.23.3
git+https://github.com/openai/CLIP.git
--extra-index-url https://download.pytorch.org/whl/cu124
torch==2.4.0
torchvision==0.19.0