zhuchaowe/qwen_agent
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

降维到128

Browse Source
This commit is contained in:
朱潮 2025-11-20 21:37:46 +08:00
parent ef7cb7560f
commit 1dd15185d6
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
embedding/manager.py
View File

@ -62,7 +62,11 @@ class GlobalModelManager:
loop = asyncio.get_event_loop() loop = asyncio.get_event_loop()
self._model = await loop.run_in_executor( self._model = await loop.run_in_executor(
None, 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 self._load_time = time.time() - start_time