gyt/opendan
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix text embedding model load error

Browse Source
This commit is contained in:
liyaxing
2023-09-27 18:53:58 +08:00
parent 87f82ec9e2
commit e858978778
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/aios_kernel/local_st_compute_node.py
+1 -1
View File
@@ -32,7 +32,7 @@ class LocalSentenceTransformer_Text_ComputeNode(Queue_ComputeNode):
try: try:
from sentence_transformers import SentenceTransformer from sentence_transformers import SentenceTransformer
self.model = SentenceTransformer(self.model) self.model = SentenceTransformer(self.model_name)
except Exception as err: except Exception as err:
logger.error(f"load model {self.model} failed: {err}") logger.error(f"load model {self.model} failed: {err}")
return False return False