torch
transformers
omegaconf
registrable
huggingface_hub
datasets
pyyaml
openai
numpy
colorama
json-repair
