kris/ForcePilot
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

add logger for loading local model

Browse Source
This commit is contained in:
Wenjie Zhang 2025-04-13 23:23:34 +08:00
parent 11f932d4b3
commit d487b4df7d
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/models/embedding.py
View File

@ -62,6 +62,8 @@ class LocalEmbeddingModel(FlagModel, BaseEmbeddingModel):
if os.path.exists(_path := os.path.join(os.getenv("MODEL_DIR"), self.model)): if os.path.exists(_path := os.path.join(os.getenv("MODEL_DIR"), self.model)):
self.model = _path 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}`") logger.info(f"Loading local model `{info['name']}` from `{self.model}` with device `{config.device}`")