arg name err patch

This commit is contained in:
binary-husky
2024-05-13 23:40:35 +08:00
parent fd2f8b9090
commit 70d5f2a7df

View File

@@ -850,7 +850,7 @@ if "deepseekcoder" in AVAIL_LLM_MODELS: # deepseekcoder
if "deepseek-chat" in AVAIL_LLM_MODELS or "deepseek-coder" in AVAIL_LLM_MODELS: if "deepseek-chat" in AVAIL_LLM_MODELS or "deepseek-coder" in AVAIL_LLM_MODELS:
try: try:
deepseekapi_noui, deepseekapi_ui = get_predict_function( deepseekapi_noui, deepseekapi_ui = get_predict_function(
APIKEY="DEEPSEEK_API_KEY", max_output_token=4096, disable_proxy=False api_key_conf_name="DEEPSEEK_API_KEY", max_output_token=4096, disable_proxy=False
) )
model_info.update({ model_info.update({
"deepseek-chat":{ "deepseek-chat":{