diff --git a/setup.py b/setup.py index bf361a62..4d3265e2 100644 --- a/setup.py +++ b/setup.py @@ -31,7 +31,7 @@ extras = {} -extras["st"] = ["sentence_transformers==2.7.0"] +extras["st"] = ["sentence_transformers==3.1.1"] extras["diffusers"] = ["diffusers==0.30.0", "accelerate==0.33.0"] # Includes `peft` as PEFT requires `torch` so having `peft` as a core dependency # means that `torch` will be installed even if the `torch` extra is not specified.