From 8069c8caa9833a05b587d983c309aa7e3fe8ca00 Mon Sep 17 00:00:00 2001
From: Vladimir Mandic <mandic00@live.com>
Date: Fri, 28 Mar 2025 21:29:18 -0400
Subject: [PATCH] add llm to progress

Signed-off-by: Vladimir Mandic <mandic00@live.com>
---
 scripts/prompt_enhance.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/scripts/prompt_enhance.py b/scripts/prompt_enhance.py
index a218e081d..51ebd475e 100644
--- a/scripts/prompt_enhance.py
+++ b/scripts/prompt_enhance.py
@@ -205,6 +205,7 @@ class Script(scripts.Script):
         p.negative_prompt = shared.prompt_styles.apply_negative_styles_to_prompt(p.negative_prompt, p.styles)
         shared.prompt_styles.apply_styles_to_extra(p)
         p.styles = []
+        shared.state.begin('LLM')
         p.prompt = self.enhance(
             prompt=p.prompt,
             model=llm_model,
@@ -214,3 +215,4 @@ class Script(scripts.Script):
             temperature=temperature,
             penalty=repetition_penalty,
         )
+        shared.state.end()