oobabooga/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2026-04-05 22:55:22 +00:00
Code Issues Projects Releases Packages Wiki Activity

Fix stopping string detection for chromadb/context-1

Browse source
This commit is contained in:
oobabooga 2026-03-28 06:09:53 -07:00
parent 4cbea02ed4
commit bda95172bd
1 changed files with 4 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

5
modules/chat.py
View file

@ -671,7 +671,10 @@ def get_stopping_strings(state):
# Handle GPT-OSS as a special case
if '<|channel|>final<|message|>' in state['instruction_template_str'] and "<|end|>" in result:
result.remove("<|end|>")
result.append("<|result|>")
if '<|result|>' in state['instruction_template_str']:
result.append("<|result|>")
elif '<|return|>' in state['instruction_template_str']:
result.append("<|return|>")
result = list(set(result))
if shared.args.verbose: