diff --git a/pyproject.toml b/pyproject.toml
index 547c363f..6016864e 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -48,7 +48,7 @@ dependencies = [
 extras = [
     # Heavy dependencies that aren't for everyday use
     "outlines",
-    "sentence-transformers"
+    "infinity-emb"
 ]
 dev = [
     "ruff == 0.3.2"