datasets
accelerate
einops
transformers>=4.33.0
safetensors>=0.3.0