From e858978778b6dd6c593dbca1247be84c75422359 Mon Sep 17 00:00:00 2001
From: liyaxing <liyaxing@buckyos.com>
Date: Wed, 27 Sep 2023 18:53:58 +0800
Subject: [PATCH] Fix text embedding model load error

---
 src/aios_kernel/local_st_compute_node.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/aios_kernel/local_st_compute_node.py b/src/aios_kernel/local_st_compute_node.py
index d4cd17e..9afa6b4 100644
--- a/src/aios_kernel/local_st_compute_node.py
+++ b/src/aios_kernel/local_st_compute_node.py
@@ -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