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

add image insert/query in knowledge base

Browse Source
This commit is contained in:
tsukasa
2023-09-28 16:20:30 +08:00
parent e0c4eb5849
commit 56c0202552
3 changed files with 47 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/aios_kernel/local_st_compute_node.py
+2 -2
View File
@@ -146,7 +146,7 @@ class LocalSentenceTransformer_Image_ComputeNode(Queue_ComputeNode):
return None
file_size = image_obj.get_file_size()
print(f"got image object: {source.to_base58()}, size: {file_size}")
# print(f"got image object: {source.to_base58()}, size: {file_size}")
image_data = (
KnowledgeStore()
@@ -207,7 +207,7 @@ class LocalSentenceTransformer_Image_ComputeNode(Queue_ComputeNode):
"error": {"code": -1, "message": "load image failed"},
}
sentence_embeddings = self.model.encode(img)
sentence_embeddings = self.model.encode(img, show_progress_bar=False).tolist()
# logger.debug(f"LocalSentenceTransformer_Text_ComputeNode task sentence_embeddings: {sentence_embeddings}")
return {