From 3039aeffeb8958724de56912d0d90267b87a7074 Mon Sep 17 00:00:00 2001
From: oobabooga <112222186+oobabooga@users.noreply.github.com>
Date: Tue, 5 Aug 2025 11:35:17 -0700
Subject: [PATCH] Fix parsing the gpt-oss-20b template

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

diff --git a/modules/models_settings.py b/modules/models_settings.py
index c4dfb0ed..8ed7f953 100644
--- a/modules/models_settings.py
+++ b/modules/models_settings.py
@@ -90,7 +90,7 @@ def get_model_metadata(model):
             template = template.replace('eos_token', "'{}'".format(eos_token))
             template = template.replace('bos_token', "'{}'".format(bos_token))
 
-            template = re.sub(r'raise_exception\([^)]*\)', "''", template)
+            template = re.sub(r"\{\{-?\s*raise_exception\(.*?\)\s*-?\}\}", "", template, flags=re.DOTALL)
             template = re.sub(r'{% if add_generation_prompt %}.*', '', template, flags=re.DOTALL)
             model_settings['instruction_template'] = 'Custom (obtained from model metadata)'
             model_settings['instruction_template_str'] = template
@@ -149,7 +149,7 @@ def get_model_metadata(model):
 
                     template = template.replace(k, "'{}'".format(value))
 
-            template = re.sub(r'raise_exception\([^)]*\)', "''", template)
+            template = re.sub(r"\{\{-?\s*raise_exception\(.*?\)\s*-?\}\}", "", template, flags=re.DOTALL)
             template = re.sub(r'{% if add_generation_prompt %}.*', '', template, flags=re.DOTALL)
             model_settings['instruction_template'] = 'Custom (obtained from model metadata)'
             model_settings['instruction_template_str'] = template