Mirrors/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2024-11-25 17:29:22 +01:00
Code Issues Packages Projects Releases Wiki Activity

Fix You You You appearing in chat mode

Browse Source
This commit is contained in:
oobabooga 2023-03-12 03:25:56 -03:00
parent bcf0075278
commit e2da6b9685
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
modules/chat.py
View File

@ -84,6 +84,7 @@ def extract_message_from_reply(question, reply, name1, name2, check, impersonate
tmp = f"\n{asker}:" tmp = f"\n{asker}:"
for j in range(1, len(tmp)): for j in range(1, len(tmp)):
if reply[-j:] == tmp[:j]: if reply[-j:] == tmp[:j]:
reply = reply[:-j]
substring_found = True substring_found = True
return reply, next_character_found, substring_found return reply, next_character_found, substring_found