torch
transformers
pillow
sentencepiece