zhengjl/text-generation-webui
Public Access
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Improve pygmalion line breaks

Browse Source
This commit is contained in:
oobabooga
2023-01-14 23:26:14 -03:00
parent 830ec7123a
commit 9a7f187b5a
2 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
server.py
View File

@@ -201,10 +201,11 @@ elif args.chat:
reply = reply[len(question):].split('\n')[0].strip() reply = reply[len(question):].split('\n')[0].strip()
else: else:
reply = generate_reply(question, tokens, inference_settings, selected_model)[0] reply = generate_reply(question, tokens, inference_settings, selected_model)[0]
reply = reply[len(question):].strip() reply = reply[len(question):]
idx = reply.find(f"\n{name1}:") idx = reply.find(f"\n{name1}:")
if idx != -1: if idx != -1:
reply = reply[:idx] reply = reply[:idx]
reply = reply.replace('\n', '\n\n').strip()
history.append((text, reply)) history.append((text, reply))
return history return history

BIN
webui.png
View File

Binary file not shown.
Side by Side

Before

Width:  |  Height:  |  Size: 153 KiB