From 2ce448214663dba1dcdcae81c917cb2ea247d40c Mon Sep 17 00:00:00 2001 From: binary-husky Date: Sat, 18 May 2024 15:47:25 +0800 Subject: [PATCH] fix new ModelOverride fn bug --- request_llms/bridge_all.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/request_llms/bridge_all.py b/request_llms/bridge_all.py index 9a12f9845..d5057ceae 100644 --- a/request_llms/bridge_all.py +++ b/request_llms/bridge_all.py @@ -1011,7 +1011,7 @@ def mutex_manager(window_mutex, observe_window): import core_functional def execute_model_override(llm_kwargs, additional_fn, method): functional = core_functional.get_core_functions() - if 'ModelOverride' in functional[additional_fn]: + if (additional_fn in functional) and 'ModelOverride' in functional[additional_fn]: # 热更新Prompt & ModelOverride importlib.reload(core_functional) functional = core_functional.get_core_functions()