gyt/opendan
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

local text embedding ok

Browse Source
This commit is contained in:
tsukasa
2023-09-27 20:50:53 +08:00
parent e858978778
commit 5b6de8b82b
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/aios_kernel/local_st_compute_node.py
+1 -1
View File
@@ -58,7 +58,7 @@ class LocalSentenceTransformer_Text_ComputeNode(Queue_ComputeNode):
logger.debug( logger.debug(
f"LocalSentenceTransformer_Text_ComputeNode task input: {input}" 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}") # logger.debug(f"LocalSentenceTransformer_Text_ComputeNode task sentence_embeddings: {sentence_embeddings}")
return { return {
"state": ComputeTaskState.DONE, "state": ComputeTaskState.DONE,