Fix text embedding model load error

2023-09-27 18:53:58 +08:00
parent 87f82ec9e2
commit e858978778
1 changed files with 1 additions and 1 deletions
@@ -32,7 +32,7 @@ class LocalSentenceTransformer_Text_ComputeNode(Queue_ComputeNode):
        try:
            from sentence_transformers import SentenceTransformer

-            self.model = SentenceTransformer(self.model)
+            self.model = SentenceTransformer(self.model_name)
        except Exception as err:
            logger.error(f"load model {self.model} failed: {err}")
            return False