transformers
einops
accelerate
sentencepiece