torch==1.9.0 torchvision==0.10.0 transformers==4.9.2 accelerate==0.5.1 numpy==1.21.2 scikit-learn==0.24.2 termcolor==1.1.0 requests==2.26.0 transformers accelerate psutil termcolor