Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add Refresh Buttons for Checkpoint/VAE Dropdowns #761

Open
wants to merge 8 commits into
base: dev
Choose a base branch
from
Open

Add Refresh Buttons for Checkpoint/VAE Dropdowns #761

Changes from all commits
Commits
Show all changes
8 commits
Select commit Hold shift + click to select a range
84bec67
add refresh buttons for ckpt/vae dropdowns
MisterChief95 Jan 2, 2025
3b7f188
add refresh buttons for ckpt/vae dropdowns
MisterChief95 Jan 2, 2025
dc25a75
add refresh buttons for ckpt/vae dropdowns
MisterChief95 Jan 2, 2025
833d397
add refresh buttons for ckpt/vae dropdowns
MisterChief95 Jan 2, 2025
ddfd1e2
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] Jan 2, 2025
7142287
Use iterable unpacking for refreshed_args
MisterChief95 Jan 2, 2025
59c29e5
Use iterable unpacking for refreshed_args
MisterChief95 Jan 2, 2025
3802f16
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] Jan 2, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
52 changes: 38 additions & 14 deletions aaaaaa/ui.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -12,6 +12,9 @@
from adetailer import ADETAILER, __version__
from adetailer.args import ALL_ARGS, MASK_MERGE_INVERT
from controlnet_ext import controlnet_exists, controlnet_type, get_cn_models
from modules.sd_models import checkpoint_tiles, list_models
from modules.shared_items import refresh_vae_list, sd_vae_items
from modules.ui_common import create_refresh_button

if controlnet_type == "forge":
from lib_controlnet import global_state
Expand Down Expand Up @@ -526,13 +529,25 @@ def inpainting(w: Widgets, n: int, is_img2img: bool, webui_info: WebuiInfo): #

ckpts = ["Use same checkpoint", *webui_info.checkpoints_list]

w.ad_checkpoint = gr.Dropdown(
label="ADetailer checkpoint" + suffix(n),
choices=ckpts,
value=ckpts[0],
visible=True,
elem_id=eid("ad_checkpoint"),
)
with gr.Row():
w.ad_checkpoint = gr.Dropdown(
label="ADetailer checkpoint" + suffix(n),
choices=ckpts,
value=ckpts[0],
visible=True,
elem_id=eid("ad_checkpoint"),
)
create_refresh_button(
refresh_component=w.ad_checkpoint,
refresh_method=list_models,
refreshed_args=lambda: {
"choices": [
"Use same checkpoint",
*checkpoint_tiles(use_short=True),
]
},
elem_id="ad_checkpoint_refresh",
)

with gr.Column(variant="compact"):
w.ad_use_vae = gr.Checkbox(
Expand All @@ -544,13 +559,22 @@ def inpainting(w: Widgets, n: int, is_img2img: bool, webui_info: WebuiInfo): #

vaes = ["Use same VAE", *webui_info.vae_list]

w.ad_vae = gr.Dropdown(
label="ADetailer VAE" + suffix(n),
choices=vaes,
value=vaes[0],
visible=True,
elem_id=eid("ad_vae"),
)
with gr.Row():
w.ad_vae = gr.Dropdown(
label="ADetailer VAE" + suffix(n),
choices=vaes,
value=vaes[0],
visible=True,
elem_id=eid("ad_vae"),
)
create_refresh_button(
refresh_component=w.ad_vae,
refresh_method=refresh_vae_list,
refreshed_args=lambda: {
"choices": ["Use same VAE", *sd_vae_items()]
},
elem_id="ad_vae_refresh",
)

with gr.Row(), gr.Column(variant="compact"):
w.ad_use_sampler = gr.Checkbox(
Expand Down