diff --git a/setup.py b/setup.py
index ead947d01c..4249e21924 100644
--- a/setup.py
+++ b/setup.py
@@ -35,7 +35,7 @@
     "accelerate >= 0.33.0, < 0.34.0",
     "diffusers == 0.29.2",
     "huggingface_hub >= 0.24.7",
-    "sentence-transformers[train] == 3.0.1",
+    "sentence-transformers == 3.2.1",
 ]
 
 TESTS_REQUIRE = [