From 5b6de8b82be1bc895ba492ca5d3a7aad9e9bc27c Mon Sep 17 00:00:00 2001 From: tsukasa Date: Wed, 27 Sep 2023 20:50:53 +0800 Subject: [PATCH] local text embedding ok --- 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 9afa6b4..43c2114 100644 --- a/src/aios_kernel/local_st_compute_node.py +++ b/src/aios_kernel/local_st_compute_node.py @@ -58,7 +58,7 @@ class LocalSentenceTransformer_Text_ComputeNode(Queue_ComputeNode): logger.debug( f"LocalSentenceTransformer_Text_ComputeNode task input: {input}" ) - sentence_embeddings = self.model.encode(input) + sentence_embeddings = self.model.encode(input).tolist() # logger.debug(f"LocalSentenceTransformer_Text_ComputeNode task sentence_embeddings: {sentence_embeddings}") return { "state": ComputeTaskState.DONE,