datasets
transformers
accelerate
einops
safetensors