transformers==4.40.0
datasets
pillow
numpy
torch
flash-attn