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:
|
||||
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:
|
||||
|
||||
@@ -136,4 +136,4 @@ aiosqlite
|
||||
python-telegram-bot
|
||||
pydub
|
||||
stability_sdk
|
||||
sentence-transformers==2.2
|
||||
sentence-transformers==2.2.2
|
||||
|
||||
Reference in New Issue
Block a user