diff --git a/modules/ui_model_menu.py b/modules/ui_model_menu.py index b7a67a8369..2525cae617 100644 --- a/modules/ui_model_menu.py +++ b/modules/ui_model_menu.py @@ -239,6 +239,7 @@ def load_model_wrapper(selected_model, loader, autoload=False): if 'instruction_template' in settings: output += '\n\nIt seems to be an instruction-following model with template "{}". In the chat tab, instruct or chat-instruct modes should be used.'.format(settings['instruction_template']) + output += '\n\n Starting warmup ...' yield output else: yield f"Failed to load `{selected_model}`."