github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-09-05 02:51:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: default proxyllm generator function (#971)

Browse Source
This commit is contained in:
xtyuns
2023-12-25 20:00:04 +08:00
committed by GitHub
parent 0c46c339ca
commit 048fb6c402
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
dbgpt/model/llm_out/proxy_llm.py
View File

@@ -34,7 +34,7 @@ def proxyllm_generate_stream(
model_name = model_params.model_name
default_error_message = f"{model_name} LLM is not supported"
generator_function = generator_mapping.get(
model_name, lambda: default_error_message
model_name, lambda *args: [default_error_message]
)
yield from generator_function(model, tokenizer, params, device, context_len)