torch
torchvision
scikit-learn
matplotlib
Pillow
sentencepiece==0.2.0
numpy
fast_pytorch_kmeans
psutil
transformers