torch>=1.7.0
transformers>=4.0.0
scikit-learn>=0.24.0
