--index-url https://pypi.tuna.tsinghua.edu.cn/simple
numpy
fire
transformers>=4.32.1
torch==2.0.1
sentencepiece
tokenizers==0.13.3
wandb
accelerate
protobuf