local text embedding ok

2023-09-27 20:50:53 +08:00
parent e858978778
commit 5b6de8b82b
1 changed files with 1 additions and 1 deletions
@@ -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,