torch
transformers
omegaconf
registrable
huggingface_hub
datasets
pyyaml
openai
evaluate
numpy
rouge_scorer