fix error

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

No files matched your search

+1 -1
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"]
+1 -1
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"]