Add missing parameters in model_chat.py

[Choiuijin1125]

In model/model_training/tools/model_chat.py, process_output function supposed to have a method and bot_name parameters

Function call line

Open-Assistant/model/model_training/tools/model_chat.py

Line 144 in 750c59d

reply = process_output(output, method, bot_name)

Definition

Open-Assistant/model/model_training/tools/model_chat.py

Lines 89 to 102 in 750c59d

	def process_output(output):
	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]
	elif method == "v2.5":
	answer = output.split(f"{ChatRole.assistant}")[-1]
	# answer = answer.split("</s>")[0].replace(SeqToken.end, "").lstrip()
	# elif method == "v3":
	# answer = output.split(f"{SeqToken.begin}{ChatRole.assistant}{SeqToken.delimiter}")[-1]
	# answer = answer.split("</s>")[0].replace(SeqToken.end, "").lstrip()
	else:
	answer = output.split("\n\n{}:".format(bot_name))[-1]
	answer = answer.split("</s>")[0].replace("<|endoftext|>", "").lstrip().split("\n\n{}:".format(bot_name))[0]
	return answer

[sanagno]

Thanks!