OPENGIT/stable-diffusion-webui
1
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2025-02-22 22:25:02 +08:00
Code Issues Packages Projects Releases Wiki Activity

Make extras work again

Browse Source 
Not all postprocessing scripts work
This commit is contained in:
missionfloyd 2023-12-17 00:43:18 -07:00
parent a8e41f585e
commit 5b636b3105
2 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
modules/postprocessing.py
View File

@ -13,6 +13,9 @@ def run_postprocessing(id_task, extras_mode, image, image_folder, input_dir, out
outputs = [] outputs = []
if isinstance(image, dict):
image = image["composite"]
def get_images(extras_mode, image, image_folder, input_dir): def get_images(extras_mode, image, image_folder, input_dir):
if extras_mode == 1: if extras_mode == 1:
for img in image_folder: for img in image_folder:

4
modules/ui_postprocessing.py
View File

@ -4,14 +4,14 @@ import modules.generation_parameters_copypaste as parameters_copypaste
def create_ui(): def create_ui():
dummy_component = gr.Label(visible=False) dummy_component = gr.Textbox(visible=False)
tab_index = gr.State(value=0) tab_index = gr.State(value=0)
with gr.Row(equal_height=False, variant='compact'): with gr.Row(equal_height=False, variant='compact'):
with gr.Column(variant='compact'): with gr.Column(variant='compact'):
with gr.Tabs(elem_id="mode_extras"): with gr.Tabs(elem_id="mode_extras"):
with gr.TabItem('Single Image', id="single_image", elem_id="extras_single_tab") as tab_single: with gr.TabItem('Single Image', id="single_image", elem_id="extras_single_tab") as tab_single:
extras_image = gr.Image(label="Source", source="upload", interactive=True, type="pil", elem_id="extras_image") extras_image = gr.ImageEditor(label="Source", interactive=True, type="pil", elem_id="extras_image")
with gr.TabItem('Batch Process', id="batch_process", elem_id="extras_batch_process_tab") as tab_batch: with gr.TabItem('Batch Process', id="batch_process", elem_id="extras_batch_process_tab") as tab_batch:
image_batch = gr.Files(label="Batch Process", interactive=True, elem_id="extras_image_batch") image_batch = gr.Files(label="Batch Process", interactive=True, elem_id="extras_image_batch")