mirror of
https://github.com/oobabooga/text-generation-webui.git
synced 2024-10-01 01:26:03 -04:00
Improve pygmalion line breaks
This commit is contained in:
parent
830ec7123a
commit
9a7f187b5a
@ -201,10 +201,11 @@ elif args.chat:
|
||||
reply = reply[len(question):].split('\n')[0].strip()
|
||||
else:
|
||||
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}:")
|
||||
if idx != -1:
|
||||
reply = reply[:idx]
|
||||
reply = reply.replace('\n', '\n\n').strip()
|
||||
|
||||
history.append((text, reply))
|
||||
return history
|
||||
|
Loading…
Reference in New Issue
Block a user