Added list of supported models

goldpulpy · Oct 9, 2024 · 93305cb · 93305cb
1 parent 12d1c4c
commit 93305cb
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/pysentence_similarity/_support_models.py b/pysentence_similarity/_support_models.py
@@ -0,0 +1,14 @@
+"""Support models list."""
+
+_support_models = [
+    "all-MiniLM-L6-v2",
+    "paraphrase-MiniLM-L6-v2",
+    "all-MiniLM-L12-v2",
+    "gte-small",
+    "all-mpnet-base-v2",
+    "paraphrase-multilingual-MiniLM-L12-v2"
+    "text2vec-base-multilingual",
+    "gte-multilingual-base",
+    "gte-large",
+    "LaBSE"
+]