From 93305cbeb6d8bff29d1e608224858c4ebd6596c1 Mon Sep 17 00:00:00 2001
From: goldpulpy <sawsani1928@gmail.com>
Date: Wed, 9 Oct 2024 04:04:46 +0300
Subject: [PATCH] Added list of supported models

---
 pysentence_similarity/_support_models.py | 14 ++++++++++++++
 1 file changed, 14 insertions(+)
 create mode 100644 pysentence_similarity/_support_models.py

diff --git a/pysentence_similarity/_support_models.py b/pysentence_similarity/_support_models.py
new file mode 100644
index 0000000..9d4dd69
--- /dev/null
+++ 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"
+]