tqdm
scikit-learn
scipy<1.6,>=1.5.4
transformers
torch
numpy<1.20,>=1.19.5
setuptools
