From c51c9b152dee1eb3c1a71717bb3c2b8f830325bc Mon Sep 17 00:00:00 2001 From: Wenjie Zhang Date: Fri, 22 Nov 2024 00:04:30 +0800 Subject: [PATCH] fix --- src/models/embedding.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/models/embedding.py b/src/models/embedding.py index 8a28f237..a124e0ac 100644 --- a/src/models/embedding.py +++ b/src/models/embedding.py @@ -20,13 +20,13 @@ class EmbeddingModel(FlagModel): model_name=model_info["name"], default_path=model_info.get("default_path", None)) - logger.info(f"Loading embedding model {model_info["name"]} from {model_name_or_path}") + logger.info(f"Loading embedding model {model_info['name']} from {model_name_or_path}") super().__init__(model_name_or_path, query_instruction_for_retrieval=model_info.get("query_instruction", None), use_fp16=False, **kwargs) - logger.info(f"Embedding model {model_info["name"]} loaded") + logger.info(f"Embedding model {model_info['name']} loaded") class Reranker(FlagReranker):