add logger for loading local model

2025-04-13 23:23:34 +08:00 · 2025-04-13 23:23:34 +08:00 · d487b4df7d
commit d487b4df7d
parent 11f932d4b3
1 changed files with 2 additions and 0 deletions
--- a/src/models/embedding.py
+++ b/src/models/embedding.py
@ -62,6 +62,8 @@ class LocalEmbeddingModel(FlagModel, BaseEmbeddingModel):

        if os.path.exists(_path := os.path.join(os.getenv("MODEL_DIR"), self.model)):
            self.model = _path
+        else:
+            logger.warning(f"Local model `{info['name']}` not found in `{self.model}`, using `{info['name']}`")

        logger.info(f"Loading local model `{info['name']}` from `{self.model}` with device `{config.device}`")