numpy
rouge_score
fire
openai
transformers>=4.28.1
torch
sentencepiece
tokenizers>=0.13.3
wandb
