numpy==1.24.1
datasets
accelerate
sentencepiece
protobuf