diff --git a/configs/phi-2/setup/requirements.txt b/configs/phi-2/setup/requirements.txt index a55d10c..27b92bf 100644 --- a/configs/phi-2/setup/requirements.txt +++ b/configs/phi-2/setup/requirements.txt @@ -7,7 +7,7 @@ accelerate==0.23.0 bitsandbytes==0.41.1 datasets==2.14.5 peft==0.5.0 -scikit-learn==1.3.1 +scikit-learn==1.5.0 sentencepiece==0.1.99 trl==0.7.2 protobuf==3.20.3