requirements.txt

accelerate==0.28.0
datasets==2.16.1
deepspeed==0.14.4
einops==0.8.1
gradio==5.34.0
numpy==1.26.1
Pillow==11.2.1
pyarrow==17.0.0
PyYAML==6.0.2
torch==2.3.0
torchvision==0.18.0
torchaudio==2.3.0
tqdm==4.66.5
transformers==4.51.3
wandb
easydict
flash_attn==2.6.2
diffusers
tabulate