Add sentence-transformer local text embedding supports

2023-09-25 16:22:15 +08:00
parent f472689ecb
commit 5f346b1bd9
2 changed files with 2 additions and 2 deletions
@@ -89,7 +89,7 @@ class LocalSentenceTransformer_ComputeNode(Queue_ComputeNode):

    def is_support(self, task: ComputeTask) -> bool:
        return task.task_type == ComputeTaskType.TEXT_EMBEDDING and (
-            not task.params["model_name"] or task.params["model_name"] == "all-MiniLM-L6-v2"
+            not task.params["model_name"] or task.params["model_name"] == "llama"
        )

    def is_local(self) -> bool: