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