GPUtil<2.0,>=1.4
numpy<2.0,>=1.19.5
PyYAML<7.0,>=6.0
scikit_learn<2.0,>=1.6
torch<3.0,>=2.5
transformers<5.0,>=4.42
