fix error

This commit is contained in:
qingxu fu 2023-04-06 17:23:26 +08:00
parent d4ed4efa03
commit 06f8094a0a
2 changed files with 2 additions and 2 deletions

View File

@ -146,7 +146,7 @@ def predict(inputs, top_p, temperature, chatbot=[], history=[], system_prompt=''
if additional_fn is not None:
import core_functional
importlib.reload(core_functional) # 热更新prompt
core_functional = core_functional.get_functions()
core_functional = core_functional.get_core_functions()
if "PreProcess" in core_functional[additional_fn]: inputs = core_functional[additional_fn]["PreProcess"](inputs) # 获取预处理函数(如果有的话)
inputs = core_functional[additional_fn]["Prefix"] + inputs + core_functional[additional_fn]["Suffix"]

View File

@ -103,7 +103,7 @@ def predict_tgui(inputs, top_p, temperature, chatbot=[], history=[], system_prom
if additional_fn is not None:
import core_functional
importlib.reload(core_functional) # 热更新prompt
core_functional = core_functional.get_functions()
core_functional = core_functional.get_core_functions()
if "PreProcess" in core_functional[additional_fn]: inputs = core_functional[additional_fn]["PreProcess"](inputs) # 获取预处理函数(如果有的话)
inputs = core_functional[additional_fn]["Prefix"] + inputs + core_functional[additional_fn]["Suffix"]