torch
transformers
sentencepiece
accelerate
protobuf