Add sentence-transformer local text embedding supports

This commit is contained in:
liyaxing
2023-09-25 16:22:15 +08:00
committed by tsukasa
parent f472689ecb
commit 5f346b1bd9
2 changed files with 2 additions and 2 deletions
+1 -1
View File
@@ -89,7 +89,7 @@ class LocalSentenceTransformer_ComputeNode(Queue_ComputeNode):
def is_support(self, task: ComputeTask) -> bool: def is_support(self, task: ComputeTask) -> bool:
return task.task_type == ComputeTaskType.TEXT_EMBEDDING and ( 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: def is_local(self) -> bool:
+1 -1
View File
@@ -136,4 +136,4 @@ aiosqlite
python-telegram-bot python-telegram-bot
pydub pydub
stability_sdk stability_sdk
sentence-transformers==2.2 sentence-transformers==2.2.2