# --extra-index-url https://download.pytorch.org/whl/cu121
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