nltk
datasets==1.6.2
torch==1.8.1
transformers==4.8.2
pytorch-lightning==1.4.9
tokenizers==0.10.3
numpy==1.21.2
scikit-learn==0.24.1