diff --git a/CHANGELOG.md b/CHANGELOG.md index 6dae657b5..24d4dff92 100644 --- a/CHANGELOG.md +++ b/CHANGELOG.md @@ -11,6 +11,7 @@ - **Anima** support for *img2img* and *inpaint* workflows - custom **VAE** loader for all pipelines *note*: vae still needs to be compatible with the model + - add prompt enhance info to image metadata - **UI** - add button to manually reorient input/output panels - all ui panels can be minimized/maximized by clicking on their header diff --git a/scripts/prompt_enhance.py b/scripts/prompt_enhance.py index 764b978ac..d3f1cbf64 100644 --- a/scripts/prompt_enhance.py +++ b/scripts/prompt_enhance.py @@ -958,6 +958,8 @@ class PromptEnhanceScript(scripts_manager.Script): shared.prompt_styles.apply_styles_to_extra(p) p.styles = [] jobid = shared.state.begin('LLM') + p.extra_generation_params['LLM'] = get_model_repo_from_display(llm_model) + p.extra_generation_params['Original'] = p.prompt p.prompt = self.enhance( prompt=p.prompt, seed=p.seed, @@ -980,5 +982,4 @@ class PromptEnhanceScript(scripts_manager.Script): keep_thinking=keep_thinking, ) timer.process.record('prompt') - p.extra_generation_params['LLM'] = llm_model shared.state.end(jobid)