github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-08-12 13:42:23 +00:00
Code Issues Packages Projects Releases Wiki Activity

update

Browse Source
This commit is contained in:
csunny 2023-05-21 16:18:33 +08:00
parent 604d269797
commit 5ec1f413b6
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
pilot/model/chatglm_llm.py
View File

@ -11,7 +11,7 @@ def chatglm_generate_stream(model, tokenizer, params, device, context_len=2048,
temperature = float(params.get("temperature", 1.0)) temperature = float(params.get("temperature", 1.0))
top_p = float(params.get("top_p", 1.0)) top_p = float(params.get("top_p", 1.0))
stop = params.get("stop", "###") stop = params.get("stop", "###")
echo = params.get("echo", True) echo = params.get("echo", False)
generate_kwargs = { generate_kwargs = {
"do_sample": True if temperature > 1e-5 else False, "do_sample": True if temperature > 1e-5 else False,