torch
transformers
omegaconf
registrable
huggingface_hub
datasets
pyyaml
openai
numpy
rouge_score
colorama
