lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-04-30 03:01:15 +00:00
Code Issues Packages Projects Releases Wiki Activity

rename

Browse Source
This commit is contained in:
lllyasviel
2024-02-04 21:05:05 -08:00
parent 8dc2cb98b4
commit 4b6cd359e9
2 changed files with 9 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
extensions-builtin/sd_forge_controlnet/lib_controlnet/controlnet_ui/controlnet_ui_group.py
View File

@@ -193,7 +193,7 @@ class ControlNetUiGroup(object):
self.batch_tab = None self.batch_tab = None
self.batch_image_dir = None self.batch_image_dir = None
self.merge_tab = None self.merge_tab = None
self.merge_gallery = None self.batch_input_gallery = None
self.merge_upload_button = None self.merge_upload_button = None
self.merge_clear_button = None self.merge_clear_button = None
self.create_canvas = None self.create_canvas = None
@@ -324,17 +324,15 @@ class ControlNetUiGroup(object):
else None, else None,
) )
with gr.Tab(label="Batch") as self.batch_tab: with gr.Tab(label="Batch Folder") as self.batch_tab:
# gr.HTML('Batch system is under maintaining now ... Please come back later ...')
self.batch_image_dir = gr.Textbox( self.batch_image_dir = gr.Textbox(
label="Input Directory", label="Input Directory",
placeholder="Leave empty to use img2img batch controlnet input directory", placeholder="Leave empty to use img2img batch controlnet input directory",
elem_id=f"{elem_id_tabname}_{tabname}_batch_image_dir", elem_id=f"{elem_id_tabname}_{tabname}_batch_image_dir",
) )
with gr.Tab(label="Multiple Images") as self.merge_tab: with gr.Tab(label="Batch Upload") as self.merge_tab:
# gr.HTML('Multi-image system is under maintaining now ... Please come back later ...') self.batch_input_gallery = gr.Gallery(
self.merge_gallery = gr.Gallery(
columns=[4], rows=[2], object_fit="contain", height="auto" columns=[4], rows=[2], object_fit="contain", height="auto"
) )
with gr.Row(): with gr.Row():
@@ -587,7 +585,7 @@ class ControlNetUiGroup(object):
self.input_mode, self.input_mode,
self.use_preview_as_input, self.use_preview_as_input,
self.batch_image_dir, self.batch_image_dir,
self.merge_gallery, self.batch_input_gallery,
self.generated_image, self.generated_image,
self.mask_image, self.mask_image,
self.enabled, self.enabled,
@@ -1060,7 +1058,7 @@ class ControlNetUiGroup(object):
self.merge_clear_button.click( self.merge_clear_button.click(
fn=lambda: [], fn=lambda: [],
inputs=[], inputs=[],
outputs=[self.merge_gallery], outputs=[self.batch_input_gallery],
).then( ).then(
fn=lambda x: gr.update(value=x + 1), fn=lambda x: gr.update(value=x + 1),
inputs=[self.update_unit_counter], inputs=[self.update_unit_counter],
@@ -1074,8 +1072,8 @@ class ControlNetUiGroup(object):
self.merge_upload_button.upload( self.merge_upload_button.upload(
upload_file, upload_file,
inputs=[self.merge_upload_button, self.merge_gallery], inputs=[self.merge_upload_button, self.batch_input_gallery],
outputs=[self.merge_gallery], outputs=[self.batch_input_gallery],
queue=False, queue=False,
).then( ).then(
fn=lambda x: gr.update(value=x + 1), fn=lambda x: gr.update(value=x + 1),

2
extensions-builtin/sd_forge_controlnet/lib_controlnet/external_code.py
View File

@@ -151,7 +151,7 @@ class UiControlNetUnit:
input_mode: InputMode = InputMode.SIMPLE input_mode: InputMode = InputMode.SIMPLE
use_preview_as_input: bool = False, use_preview_as_input: bool = False,
batch_image_dir: str = '', batch_image_dir: str = '',
merge_gallery: list = [], batch_input_gallery: list = [],
generated_image: Optional[np.ndarray] = None, generated_image: Optional[np.ndarray] = None,
mask_image: Optional[np.ndarray] = None, mask_image: Optional[np.ndarray] = None,
enabled: bool = True enabled: bool = True