From bfbbfc2361e26b03e5af9a26434391be9fd257f1 Mon Sep 17 00:00:00 2001
From: oobabooga <112222186+oobabooga@users.noreply.github.com>
Date: Tue, 5 Aug 2025 17:33:01 -0700
Subject: [PATCH] Ignore add_generation_prompt in GPT-OSS

---
 modules/models_settings.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/modules/models_settings.py b/modules/models_settings.py
index 3c068df0..e35e1c04 100644
--- a/modules/models_settings.py
+++ b/modules/models_settings.py
@@ -93,6 +93,7 @@ def get_model_metadata(model):
             template = re.sub(r"\{\{-?\s*raise_exception\(.*?\)\s*-?\}\}", "", template, flags=re.DOTALL)
             template = re.sub(r'raise_exception\([^)]*\)', "''", template)
             template = re.sub(r'{% if add_generation_prompt %}.*', '', template, flags=re.DOTALL)
+            template = re.sub(r'elif loop\.last and not add_generation_prompt', 'elif False', template)  # Handle GPT-OSS
             model_settings['instruction_template'] = 'Custom (obtained from model metadata)'
             model_settings['instruction_template_str'] = template
 
@@ -153,6 +154,7 @@ def get_model_metadata(model):
             template = re.sub(r"\{\{-?\s*raise_exception\(.*?\)\s*-?\}\}", "", template, flags=re.DOTALL)
             template = re.sub(r'raise_exception\([^)]*\)', "''", template)
             template = re.sub(r'{% if add_generation_prompt %}.*', '', template, flags=re.DOTALL)
+            template = re.sub(r'elif loop\.last and not add_generation_prompt', 'elif False', template)  # Handle GPT-OSS
             model_settings['instruction_template'] = 'Custom (obtained from model metadata)'
             model_settings['instruction_template_str'] = template