transformers>=4.44.0,<5
accelerate>=0.30.0
sentencepiece
einops