mmarch/text-generation-webui
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Fix a bug in cai mode chat

Browse source
This commit is contained in:
oobabooga 2023-01-15 19:41:25 -03:00
parent e6691bd920
commit bb1a172da0
1 changed files with 6 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

10
server.py
View file

@ -222,8 +222,12 @@ elif args.chat or args.cai_chat:
question = context+'\n\n' question = context+'\n\n'
for i in range(len(history)): for i in range(len(history)):
question += f"{name1}: {history[i][0][3:-5].strip()}\n" if args.cai_chat:
question += f"{name2}: {history[i][1][3:-5].strip()}\n" question += f"{name1}: {history[i][0].strip()}\n"
question += f"{name2}: {history[i][1].strip()}\n"
else:
question += f"{name1}: {history[i][0][3:-5].strip()}\n"
question += f"{name2}: {history[i][1][3:-5].strip()}\n"
question += f"{name1}: {text}\n" question += f"{name1}: {text}\n"
question += f"{name2}:" question += f"{name2}:"
@ -294,8 +298,6 @@ elif args.chat or args.cai_chat:
with gr.Row(): with gr.Row():
check = gr.Checkbox(value=settings['stop_at_newline'], label='Stop generating at new line character?') check = gr.Checkbox(value=settings['stop_at_newline'], label='Stop generating at new line character?')
if args.cai_chat: if args.cai_chat:
btn.click(cai_chatbot_wrapper, [textbox, length_slider, preset_menu, model_menu, name1, name2, context, check], display1, show_progress=True, api_name="textgen") btn.click(cai_chatbot_wrapper, [textbox, length_slider, preset_menu, model_menu, name1, name2, context, check], display1, show_progress=True, api_name="textgen")
textbox.submit(cai_chatbot_wrapper, [textbox, length_slider, preset_menu, model_menu, name1, name2, context, check], display1, show_progress=True) textbox.submit(cai_chatbot_wrapper, [textbox, length_slider, preset_menu, model_menu, name1, name2, context, check], display1, show_progress=True)