torch
torchvision
transformers
accelerate
numpy
scikit-learn
termcolor
requests
psutil
colorama
sentence_transformers