torch
transformers
sentencepiece
accelerate