diff --git a/src/config/__init__.py b/src/config/__init__.py index 764778d9..b8cf6119 100644 --- a/src/config/__init__.py +++ b/src/config/__init__.py @@ -102,9 +102,9 @@ class Config(SimpleConfig): # 修改为按照子元素合并 # _models = {**_models, **_models_private} - 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"]} + self.model_names = {**_models["MODEL_NAMES"], **_models_private.get("MODEL_NAMES", {})} + self.embed_model_names = {**_models["EMBED_MODEL_INFO"], **_models_private.get("EMBED_MODEL_INFO", {})} + self.reranker_names = {**_models["RERANKER_LIST"], **_models_private.get("RERANKER_LIST", {})} def _save_models_to_file(self): _models = {