Logic/rag_chat
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix(memory): specify full embedding model identifier
Some checks failed
Build and Push Docker / build-and-push (push) Failing after 15m29s
Details

Browse Source
This commit is contained in:
gameloader
2025-11-27 21:08:01 +08:00
parent a43c6a10ac
commit dcc20609c8
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
memory_module/memory_integration.py
View File

@ -210,7 +210,7 @@ class LocalMemoryIntegration:
logger.info(f"ChromaDB persistent client initialized at {chroma_path}") logger.info(f"ChromaDB persistent client initialized at {chroma_path}")
# 5. 初始化本地 Embedding 模型 (低内存占用) # 5. 初始化本地 Embedding 模型 (低内存占用)
model_name = 'bge-small-zh-v1.5' # 约 100MB 内存 model_name = 'BAAI/bge-small-zh-v1.5' # 约 100MB 内存
try: try:
logger.info(f"Loading local embedding model: {model_name}...") logger.info(f"Loading local embedding model: {model_name}...")
self.embedding_model = SentenceTransformer(model_name, device='cpu') self.embedding_model = SentenceTransformer(model_name, device='cpu')