diff --git a/jobs/bert_finetune/requirements.txt b/jobs/bert_finetune/requirements.txt index 8f349f4..f2496b4 100644 --- a/jobs/bert_finetune/requirements.txt +++ b/jobs/bert_finetune/requirements.txt @@ -1,7 +1,7 @@ accelerate==0.30.0 datasets==2.19.1 evaluate==0.4.2 -scikit-learn==1.4.2 +scikit-learn==1.5.0 torch==2.3.0 transformers==4.40.2 wandb>=0.17.0 \ No newline at end of file