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

fix: guanaco output

Browse Source
This commit is contained in:
csunny
2023-06-04 20:15:27 +08:00
parent 7adc30d6b9
commit f7fe66b5e5
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
pilot/out_parser/base.py
View File

@@ -57,7 +57,7 @@ class BaseOutputParser(ABC):
output = data["text"][skip_echo_len:].strip() output = data["text"][skip_echo_len:].strip()
elif "guanaco" in CFG.LLM_MODEL: elif "guanaco" in CFG.LLM_MODEL:
# output = data["text"][skip_echo_len + 14:].replace("<s>", "").strip() # output = data["text"][skip_echo_len + 14:].replace("<s>", "").strip()
output = data["text"][skip_echo_len:].replace("<s>", "").strip() output = data["text"][skip_echo_len + 2:].replace("<s>", "").strip()
else: else:
output = data["text"].strip() output = data["text"].strip()