transformers==4.3.2
pytorch-lightning==1.2.4
torch==1.8.0
datasets==1.3.0