fire
triton
pydantic==2.8.2
accelerate
deepspeed
huggingface-hub==0.23.4
imageio==2.25.1
numpy==1.21.6
torch==2.2.0
omegaconf==2.3.0
pandas==1.3.5
Pillow==9.4.0
tokenizers==0.13.2
transformers
xformers
safetensors==0.3.3
diffusers==0.28.2
sentencepiece==0.1.99
gradio==4.37.2