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" +]