accelerate
torch >= 1.3
datasets >= 1.8.0
sentencepiece != 0.1.92
protobuf