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}`")