From 33913825b02d20dc6cd8733495f4f8ecccf2f316 Mon Sep 17 00:00:00 2001 From: vladmandic Date: Mon, 24 Nov 2025 10:12:40 -0500 Subject: [PATCH] fix control run args Signed-off-by: vladmandic --- modules/control/run.py | 3 ++- modules/ui_control.py | 4 +++- 2 files changed, 5 insertions(+), 2 deletions(-) diff --git a/modules/control/run.py b/modules/control/run.py index e05aa65af..6d02d1ed4 100644 --- a/modules/control/run.py +++ b/modules/control/run.py @@ -588,6 +588,7 @@ def control_run(state: str = '', # pylint: disable=keyword-arg-before-vararg if sd_models.get_diffusers_task(pipe) != sd_models.DiffusersTaskType.TEXT_2_IMAGE: # force vae back to gpu if not in txt2img mode sd_models.move_model(pipe.vae, devices.device) + # init scripts p.scripts = scripts_manager.scripts_control p.script_args = input_script_args or [] if len(p.script_args) == 0: @@ -596,7 +597,7 @@ def control_run(state: str = '', # pylint: disable=keyword-arg-before-vararg p.script_args = script.init_default_script_args(p.scripts) # init override scripts - if override_script_name and override_script_args: + if override_script_name and override_script_args and len(override_script_name) > 0: selectable_scripts, selectable_script_idx = script.get_selectable_script(override_script_name, p.scripts) if selectable_scripts: for idx in range(len(override_script_args)): diff --git a/modules/ui_control.py b/modules/ui_control.py index 12799a5df..4bc95917c 100644 --- a/modules/ui_control.py +++ b/modules/ui_control.py @@ -173,6 +173,8 @@ def create_ui(_blocks: gr.Blocks=None): detailer_enabled, detailer_prompt, detailer_negative, detailer_steps, detailer_strength, detailer_resolution = shared.yolo.ui('control') with gr.Row(): + override_script_name = gr.State(value='', visible=False, elem_id='control_override_script_name') + override_script_args = gr.State(value='', visible=False, elem_id='control_override_script_args') override_settings = ui_common.create_override_inputs('control') with gr.Row(variant='compact', elem_id="control_extra_networks", elem_classes=["extra_networks_root"], visible=False) as extra_networks_ui: @@ -288,7 +290,7 @@ def create_ui(_blocks: gr.Blocks=None): enable_hr, hr_sampler_index, hr_denoising_strength, hr_resize_mode, hr_resize_context, hr_upscaler, hr_force, hr_second_pass_steps, hr_scale, hr_resize_x, hr_resize_y, refiner_steps, refiner_start, refiner_prompt, refiner_negative, video_skip_frames, video_type, video_duration, video_loop, video_pad, video_interpolate, - override_settings, + override_script_name, override_script_args, override_settings, ] output_fields = [ preview_process,