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

Fix some regexes

Browse source
This commit is contained in:
oobabooga 2023-01-19 19:59:34 -03:00
parent e61138bdad
commit 11c3214981
2 changed files with 4 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

3
server.py
View file

@ -256,9 +256,10 @@ if args.chat or args.cai_chat:
for reply in generate_reply(question, tokens, inference_settings, selected_model, eos_token=eos_token):
next_character_found = False
previous_idx = [m.start() for m in re.finditer(f"\n{name2}:", question)]
previous_idx = [m.start() for m in re.finditer(f"(^|\n){name2}:", question)]
idx = [m.start() for m in re.finditer(f"(^|\n){name2}:", reply)]
idx = idx[len(previous_idx)-1]
reply = reply[idx + len(f"\n{name2}:"):]
if check: