torch
transformers
nltk
sentencepiece