降维到128

2025-11-20 21:37:46 +08:00 · 2025-11-20 21:37:46 +08:00 · 1dd15185d6
commit 1dd15185d6
parent ef7cb7560f
1 changed files with 5 additions and 1 deletions
--- a/embedding/manager.py
+++ b/embedding/manager.py
@ -62,7 +62,11 @@ class GlobalModelManager:
                loop = asyncio.get_event_loop()
                self._model = await loop.run_in_executor(
                    None, 
-                    lambda: SentenceTransformer(model_path, device=self._device)
+                    lambda: SentenceTransformer(
+                        model_path,
+                        device=self._device,
+                        truncate_dim=128
+                    )
                )
                
                self._load_time = time.time() - start_time