sets an asymetric retrieval model as default vectorizer

justin-cechmanek · justin-cechmanek · commit d9aacf900cee · 2024-05-07T14:02:50.000-07:00
diff --git a/redisvl/extensions/session_manager/session.py b/redisvl/extensions/session_manager/session.py
@@ -65,7 +65,7 @@ def __init__(
 
         if vectorizer is None:
             self._vectorizer = HFTextVectorizer(
-                model="sentence-transformers/all-mpnet-base-v2"
+                model="sentence-transformers/msmarco-distilbert-cos-v5"
             )
 
         self.set_distance_threshold(distance_threshold)

Original file line number	Diff line number	Diff line change
`@@ -65,7 +65,7 @@ def __init__(`
`65`	`65`
`66`	`66`	`if vectorizer is None:`
`67`	`67`	`self._vectorizer = HFTextVectorizer(`
`68`		`- model="sentence-transformers/all-mpnet-base-v2"`
	`68`	`+ model="sentence-transformers/msmarco-distilbert-cos-v5"`
`69`	`69`	`)`
`70`	`70`
`71`	`71`	`self.set_distance_threshold(distance_threshold)`