mirror of
https://github.com/Bing-su/adetailer.git
synced 2026-01-26 19:29:54 +00:00
feat(script): same sampler option
This commit is contained in:
@@ -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"),
|
||||
)
|
||||
|
||||
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user