From ac0d51818b9fbf7159a8d236e1ed5f65a20a6249 Mon Sep 17 00:00:00 2001 From: RobeSantoro Date: Sun, 3 Mar 2024 19:43:01 +0100 Subject: [PATCH 1/2] fix: update unpacking of create_output_panel return value --- iz_helpers/ui.py | 18 ++++++++---------- 1 file changed, 8 insertions(+), 10 deletions(-) diff --git a/iz_helpers/ui.py b/iz_helpers/ui.py index 0c6062c..5fcbbbb 100644 --- a/iz_helpers/ui.py +++ b/iz_helpers/ui.py @@ -249,16 +249,14 @@ Our best experience and trade-off is the R-ERSGAn4x upscaler. """ ) - with gr.Column(scale=1, variant="compact"): - output_video = gr.Video(label="Output").style(width=512, height=512) - ( - out_image, - generation_info, - html_info, - html_log, - ) = create_output_panel( - "infinite-zoom", shared.opts.outdir_img2img_samples - ) + with gr.Column(scale=1, variant="compact"): + output_video = gr.Video(label="Output").style(width=512, height=512) + output_panel = create_output_panel("infinite-zoom", shared.opts.outdir_img2img_samples) + + out_image = output_panel.gallery + generation_info = output_panel.generation_info + html_info = output_panel.infotext + html_log = output_panel.html_log generate_btn.click( fn=wrap_gradio_gpu_call(create_zoom, extra_outputs=[None, "", ""]), From cfbd32363f27ad6a2d461fdeb6a1f8dc7d3315cc Mon Sep 17 00:00:00 2001 From: vahid khroasani <62482657+v8hid@users.noreply.github.com> Date: Wed, 6 Mar 2024 07:13:08 +0300 Subject: [PATCH 2/2] Apply suggestions from code review Co-authored-by: Loki's Wager <32408858+LokiWager@users.noreply.github.com> Signed-off-by: vahid khroasani <62482657+v8hid@users.noreply.github.com> --- iz_helpers/ui.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/iz_helpers/ui.py b/iz_helpers/ui.py index 5fcbbbb..7cb0430 100644 --- a/iz_helpers/ui.py +++ b/iz_helpers/ui.py @@ -253,6 +253,14 @@ Our best experience and trade-off is the R-ERSGAn4x upscaler. output_video = gr.Video(label="Output").style(width=512, height=512) output_panel = create_output_panel("infinite-zoom", shared.opts.outdir_img2img_samples) + if isinstance(output_panel, tuple): + ( + out_image, + generation_info, + html_info, + html_log, + ) = output_panel + else: out_image = output_panel.gallery generation_info = output_panel.generation_info html_info = output_panel.infotext