transformers==4.44.2
accelerate==0.31.0
deepspeed==0.14.5
numpy==1.24.3
diffusers
datasets
requests
fastapi
scipy
pandas
xformers
ftfy
Jinja2
bitsandbytes
safetensors
pyyaml
pillow==10.3.0
gradio
--extra-index-url https://download.pytorch.org/whl/cu124
torch==2.4.1
torchvision==0.19.1
torchaudio==2.4.1