From 2bfe284dd105a83f48aa0c9b6710c8640d4a0d96 Mon Sep 17 00:00:00 2001 From: Wenjie Zhang Date: Thu, 10 Apr 2025 11:41:21 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8D=E9=85=8D=E7=BD=AE=E8=A6=86?= =?UTF-8?q?=E7=9B=96=E6=9B=B4=E6=96=B0=E7=9A=84bug?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/config/__init__.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/src/config/__init__.py b/src/config/__init__.py index da6bcac8..764778d9 100644 --- a/src/config/__init__.py +++ b/src/config/__init__.py @@ -99,11 +99,12 @@ class Config(SimpleConfig): except FileNotFoundError: _models_private = {} - _models = {**_models, **_models_private} + # 修改为按照子元素合并 + # _models = {**_models, **_models_private} - self.model_names = _models["MODEL_NAMES"] - self.embed_model_names = _models["EMBED_MODEL_INFO"] - self.reranker_names = _models["RERANKER_LIST"] + self.model_names = {**_models["MODEL_NAMES"], **_models_private["MODEL_NAMES"]} + self.embed_model_names = {**_models["EMBED_MODEL_INFO"], **_models_private["EMBED_MODEL_INFO"]} + self.reranker_names = {**_models["RERANKER_LIST"], **_models_private["RERANKER_LIST"]} def _save_models_to_file(self): _models = {