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

Fix GPT-OSS channel markup leaking into UI when model skips analysis block

Browse source
This commit is contained in:
oobabooga 2026-03-14 09:08:05 -07:00
parent 09a6549816
commit cb08ba63dc
3 changed files with 16 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

2
modules/ui_image_generation.py
View file

@ -728,6 +728,8 @@ def generate_prompt_variation(state):
variation = variation.rsplit("</think>", 1)[1]
elif "<|start|>assistant<|channel|>final<|message|>" in variation:
variation = variation.rsplit("<|start|>assistant<|channel|>final<|message|>", 1)[1]
elif "<|channel|>final<|message|>" in variation:
variation = variation.rsplit("<|channel|>final<|message|>", 1)[1]
elif "</seed:think>" in variation:
variation = variation.rsplit("</seed:think>", 1)[1]