datasets==2.8.0 evaluate==0.4.0 scikit-learn==1.2.0 torch==1.12.1+cu116 transformers==4.25.1 wandb==0.13.7