transformers
sentencepiece
torch
accelerate