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