zhengjl/text-generation-webui
Public Access
1
0
Fork 0
Code Issues Pull Requests 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
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