numpy
pandas
sentencepiece
torch>=1.7.0,!=1.8.0
transformers==4.16.2
pytorch-lightning==1.5.10