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

debug: add info

Browse Source
This commit is contained in:
csunny 2023-06-01 15:25:11 +08:00
parent 7d753ac5d2
commit 876bf2a66a
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
pilot/model/guanaco_llm.py
View File

@ -5,6 +5,8 @@ from pilot.conversation import ROLE_ASSISTANT, ROLE_USER
def guanaco_generate_output(model, tokenizer, params, device, context_len=2048):
"""Fork from: https://github.com/KohakuBlueleaf/guanaco-lora/blob/main/generate.py"""
print(params)
stop = params.get("stop", "###")
messages = params["prompt"].split(stop)