torch
transformers
SentencePiece
accelerate