fix new ModelOverride fn bug

binary-husky · May 18, 2024 · 2ce4482 · 2ce4482
1 parent 4282b83
commit 2ce4482
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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()