diff --git a/hf-workshop/tweet_disaster_prediction/requirements.txt b/hf-workshop/tweet_disaster_prediction/requirements.txt index 9118851..8a07c27 100644 --- a/hf-workshop/tweet_disaster_prediction/requirements.txt +++ b/hf-workshop/tweet_disaster_prediction/requirements.txt @@ -1,7 +1,7 @@ datasets==2.14.5 evaluate==0.4.0 peft==0.5.0 -scikit-learn==1.3.1 +scikit-learn==1.5.0 torch==2.1.0 transformers==4.37.2 wandb==0.15.12