sentencepiece
einops
transformers
torch
accelerate