torch==1.9.0
transformers
joblib==1.1.0