feat(script): same sampler option

This commit is contained in:
Dowon
2024-08-03 13:05:45 +09:00
parent cdb7bd6357
commit 23e3c5d778
2 changed files with 10 additions and 2 deletions

View File

@@ -550,11 +550,16 @@ def inpainting(w: Widgets, n: int, is_img2img: bool, webui_info: WebuiInfo):
elem_id=eid("ad_use_sampler"),
)
sampler_names = [
"Use same sampler",
*webui_info.sampler_names,
]
with gr.Row():
w.ad_sampler = gr.Dropdown(
label="ADetailer sampler" + suffix(n),
choices=webui_info.sampler_names,
value=webui_info.sampler_names[0],
choices=sampler_names,
value=sampler_names[1],
visible=True,
elem_id=eid("ad_sampler"),
)

View File

@@ -374,7 +374,10 @@ class AfterDetailerScript(scripts.Script):
def get_sampler(self, p, args: ADetailerArgs) -> str:
if args.ad_use_sampler:
if args.ad_sampler == "Use same sampler":
return p.sampler_name
return args.ad_sampler
if hasattr(p, "_ad_orig"):
return p._ad_orig.sampler_name
return p.sampler_name