diff --git a/embedding/manager.py b/embedding/manager.py index 25c4d28..14f301c 100644 --- 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