--extra-index-url https://download.pytorch.org/whl/cpu
torch
torchvision
scikit-learn
matplotlib
Pillow
sentencepiece
numpy
fast_pytorch_kmeans
transformers