From d487b4df7d806a659ed0bd4f2fc7005c02d6e4fc Mon Sep 17 00:00:00 2001 From: Wenjie Zhang Date: Sun, 13 Apr 2025 23:23:34 +0800 Subject: [PATCH] add logger for loading local model --- src/models/embedding.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/src/models/embedding.py b/src/models/embedding.py index aa035bc8..f956c3c5 100644 --- 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}`")