diff --git a/pyproject.toml b/pyproject.toml index 1267342fd..e27580062 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -48,7 +48,7 @@ ml = [ "colour-science>=0.4.2", "psutil>=5.9.0", "memory-profiler>=0.61.0", - "sentence-transformers>=2.2.0,<3", + "sentence-transformers>=2.2.0,<6", "accelerate>=0.20.0,<2", ] # CoreML acceleration (optional, Apple Silicon only)