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

Update server.py

Browse source
This commit is contained in:
oobabooga 2023-01-31 08:14:16 -03:00 • committed by GitHub
parent a85bb5e9a2
commit 001ecf95b2
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
server.py
View file

@ -364,6 +364,8 @@ if args.chat or args.cai_chat:
def extract_message_from_reply(question, reply, current, other, check, extensions=False):
next_character_found = False
substring_found = False
previous_idx = [m.start() for m in re.finditer(f"(^|\n){current}:", question)]
idx = [m.start() for m in re.finditer(f"(^|\n){current}:", reply)]
idx = idx[len(previous_idx)-1]
@ -373,8 +375,6 @@ if args.chat or args.cai_chat:
else:
reply = reply[idx + 1 + len(f"{current}:"):]
substring_found = False
if check:
reply = reply.split('\n')[0].strip()
else: