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

Obtain stopping strings in chat mode

Browse source
This commit is contained in:
oobabooga 2025-08-25 18:22:08 -07:00
parent ded6c41cf8
commit b657be7381
1 changed files with 11 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

20
modules/chat.py
View file

@ -456,15 +456,17 @@ def get_stopping_strings(state):
] ]
for text in texts: for text in texts:
text = text.strip() stripped_text = text.strip()
if text.startswith("<") and ">" in text: if stripped_text.startswith("<") and ">" in stripped_text:
stopping_strings.append(text.split(">")[0] + ">") stopping_strings.append(stripped_text.split(">")[0] + ">")
elif text.startswith("[") and "]" in text: elif stripped_text.startswith("[") and "]" in stripped_text:
stopping_strings.append(text.split("]")[0] + "]") stopping_strings.append(stripped_text.split("]")[0] + "]")
elif text.startswith("(") and ")" in text: elif stripped_text.startswith("(") and ")" in stripped_text:
stopping_strings.append(text.split(")")[0] + ")") stopping_strings.append(stripped_text.split(")")[0] + ")")
elif text.startswith("{") and "}" in text: elif stripped_text.startswith("{") and "}" in stripped_text:
stopping_strings.append(text.split("}")[0] + "}") stopping_strings.append(stripped_text.split("}")[0] + "}")
elif ":" in text:
stopping_strings.append(text.split(":")[0] + ":")
if 'stopping_strings' in state and isinstance(state['stopping_strings'], list): if 'stopping_strings' in state and isinstance(state['stopping_strings'], list):
stopping_strings += state.pop('stopping_strings') stopping_strings += state.pop('stopping_strings')