numpy
packaging
torch>=1.6
transformers
tqdm
