transformers==4.3.0
torch==1.6.0
matplotlib
seaborn
numpy