Add sentence-transformer local text embedding supports
This commit is contained in:
@@ -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:
|
||||||
|
|||||||
@@ -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
|
||||||
|
|||||||
Reference in New Issue
Block a user