default text embedding with local node

2023-09-26 16:37:30 +08:00
parent 411ab61880
commit cf2d917404
3 changed files with 14 additions and 14 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"] == "llama"
+            not task.params["model_name"] or task.params["model_name"] == "all-MiniLM-L6-v2"
        )

    def is_local(self) -> bool: