torch==1.7.1
transformers==4.8.2
datasets==1.7.0
numpy==1.20.3
scikit_learn==0.24.2
