oobabooga/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2026-01-03 07:10:11 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix "continue" missing an initial space in chat-instruct/chat modes

Browse source
This commit is contained in:
oobabooga 2025-10-11 17:00:25 -07:00
parent c7dd920dc8
commit 655c3e86e3
1 changed files with 3 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
modules/chat.py
View file

@ -880,7 +880,9 @@ def chatbot_wrapper(text, state, regenerate=False, _continue=False, loading_mess
# Extract the reply # Extract the reply
if state['mode'] in ['chat', 'chat-instruct']: if state['mode'] in ['chat', 'chat-instruct']:
reply = reply.lstrip() if not _continue:
reply = reply.lstrip()
if reply.startswith(state['name2'] + ':'): if reply.startswith(state['name2'] + ':'):
reply = reply[len(state['name2'] + ':'):] reply = reply[len(state['name2'] + ':'):]
elif reply.startswith(state['name1'] + ':'): elif reply.startswith(state['name1'] + ':'):