numpy
rouge_score
fire
openai
transformers>=4.28.1
torch
sentencepiece
tokenizers>=0.13.3
tqdm
scikit-learn
wandb
