torch
transformers
sentencepiece
nltk