From 05fced7395a16f9f27e7ba9f0faec1b0b6abd980 Mon Sep 17 00:00:00 2001 From: hypercryptoman <110080854+hypercryptoman@users.noreply.github.com> Date: Mon, 19 May 2025 13:53:24 +1000 Subject: [PATCH] Update prompt_enhance.py --- scripts/prompt_enhance.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/scripts/prompt_enhance.py b/scripts/prompt_enhance.py index 45e2ab46f..118695feb 100644 --- a/scripts/prompt_enhance.py +++ b/scripts/prompt_enhance.py @@ -1,4 +1,3 @@ - from dataclasses import dataclass import io import os @@ -446,7 +445,8 @@ class Script(scripts.Script): model_file = gr.Textbox(label='Model file', value=None, interactive=True, elem_id='prompt_enhance_model_file', placeholder='Optional GGUF model file inside GGUF model repo') with gr.Row(): custom_btn = gr.Button(value='Load custom model', elem_id='prompt_enhance_custom_load', variant='secondary') - custom_btn.click(fn=self.load, inputs=[model_file, model_repo, model_gguf, model_type, None], outputs=[]) + # THIS IS THE CORRECTED LINE: + custom_btn.click(fn=self.load, inputs=[model_repo, model_repo, model_gguf, model_type, model_file], outputs=[]) llm_model.change(fn=self.get_custom, inputs=[llm_model], outputs=[model_repo, model_gguf, model_type, model_file]) gr.HTML('
') with gr.Accordion('Options', open=False, elem_id='prompt_enhance_options'): @@ -515,4 +515,4 @@ class Script(scripts.Script): ) p.extra_generation_params['LLM'] = llm_model shared.state.end() - +