numpy
fire
transformers>=4.32.1
torch==2.0.1
sentencepiece
tokenizers==0.13.3
wandb
accelerate
protobuf
