diff --git a/pyproject.toml b/pyproject.toml index f8150ad..bd97c44 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -22,7 +22,7 @@ dependencies = [ "transformers==4.38.0", "accelerate==0.21.0", "bitsandbytes==0.41.0", - "scikit-learn==1.2.2", + "scikit-learn==1.5.0", "sentencepiece==0.1.99", "einops==0.6.1", "einops-exts==0.0.4", "timm==0.6.13", "gradio_client==0.2.9"