numpy==1.26.0 torch==2.1.0 transformers==4.34.0 scikit-learn==1.3.1