gensim
numpy
scikit-learn
pandas
jieba
pytorch>=1.7
transformers==4.6.1
pickle