diff --git a/requirements.txt b/requirements.txt index 3a839d8a9a..1dd27b981a 100644 --- a/requirements.txt +++ b/requirements.txt @@ -40,7 +40,7 @@ numpy>=1.24.4,<=2.0.1 # qlora things evaluate==0.4.1 scipy -scikit-learn==1.4.2 +scikit-learn==1.5.1 nvidia-ml-py==12.560.30 art tensorboard