Skip to content

Commit

Permalink
Add missing parameters in model_chat.py (#3233)
Browse files Browse the repository at this point in the history
  • Loading branch information
Choiuijin1125 and choiuijin1125 committed May 26, 2023
1 parent 750c59d commit 812d1b8
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion model/model_training/tools/model_chat.py
Expand Up @@ -86,7 +86,7 @@ def talk(human_input: str, history: List[Tuple[str, str]], sep_token: str, prefi
return prefix


def process_output(output):
def process_output(output, method, bot_name):
if method == "v2":
answer = output.split(QA_SPECIAL_TOKENS["Answer"])[-1]
answer = answer.split("</s>")[0].replace("<|endoftext|>", "").lstrip().split(QA_SPECIAL_TOKENS["Answer"])[0]
Expand Down

0 comments on commit 812d1b8

Please sign in to comment.