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