update requirements

raphaelsty · Jun 1, 2024 · 1abef0a · 1abef0a
1 parent 5e26f57
commit 1abef0a
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -14,10 +14,11 @@
     "tqdm >= 4.62.3",
     "scipy >= 1.7.3",
     "lenlp == 1.1.0",
+    "sentence-transformers >= 3.0.0",
 ]
 
-cpu = ["sentence-transformers >= 2.2.2", "faiss-cpu >= 1.7.4"]
-gpu = ["sentence-transformers >= 2.2.2", "faiss-gpu >= 1.7.4"]
+cpu = ["sentence-transformers >= 3.0.0", "faiss-cpu >= 1.7.4"]
+gpu = ["sentence-transformers >= 3.0.0", "faiss-gpu >= 1.7.4"]
 dev = [
     "numpydoc >= 1.4.0",
     "mkdocs_material >= 8.3.5",