Bing-su/adetailer
1
0
Fork 0
mirror of https://github.com/Bing-su/adetailer.git synced 2026-04-27 01:39:10 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: add separate steps

Browse Source
This commit is contained in:
Bingsu
2023-05-12 12:59:17 +09:00
parent 7ea6e4dbc7
commit 781494ce1f
2 changed files with 52 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

55
scripts/!adetailer.py
View File

@@ -210,22 +210,22 @@ class AfterDetailerScript(scripts.Script):
with gr.Group():
with gr.Row():
w[n].ad_inpaint_full_res = gr.Checkbox(
label="Inpaint at full resolution " + suffix(n),
value=True,
visible=True,
)
w[n].ad_inpaint_full_res_padding = gr.Slider(
label="Inpaint at full resolution padding, pixels "
+ suffix(n),
minimum=0,
maximum=256,
step=4,
value=0,
visible=True,
)
with gr.Column():
w[n].ad_inpaint_full_res = gr.Checkbox(
label="Inpaint at full resolution " + suffix(n),
value=True,
visible=True,
)
w[n].ad_inpaint_full_res_padding = gr.Slider(
label="Inpaint at full resolution padding, pixels "
+ suffix(n),
minimum=0,
maximum=256,
step=4,
value=0,
visible=True,
)
with gr.Row():
with gr.Column():
w[n].ad_use_inpaint_width_height = gr.Checkbox(
label="Use separate width/height" + suffix(n),
@@ -251,6 +251,23 @@ class AfterDetailerScript(scripts.Script):
visible=True,
)
with gr.Row():
with gr.Column():
w[n].ad_use_steps = gr.Checkbox(
label="Use separate steps" + suffix(n),
value=False,
visible=True,
)
w[n].ad_steps = gr.Slider(
label="ADetailer steps" + suffix(n),
minimum=1,
maximum=150,
step=1,
value=28,
visible=True,
)
with gr.Column():
w[n].ad_use_cfg_scale = gr.Checkbox(
label="Use separate CFG scale" + suffix(n),
@@ -445,6 +462,11 @@ class AfterDetailerScript(scripts.Script):
return width, height
def get_steps(self, p, args: ADetailerArgs) -> int:
if args.ad_use_steps:
return args.ad_steps
return p.steps
def get_cfg_scale(self, p, args: ADetailerArgs) -> float:
if args.ad_use_cfg_scale:
return args.ad_cfg_scale
@@ -491,6 +513,7 @@ class AfterDetailerScript(scripts.Script):
prompt, negative_prompt = self.get_prompt(p, args)
seed, subseed = self.get_seed(p)
width, height = self.get_width_height(p, args)
steps = self.get_steps(p, args)
cfg_scale = self.get_cfg_scale(p, args)
sampler_name = p.sampler_name
@@ -521,7 +544,7 @@ class AfterDetailerScript(scripts.Script):
sampler_name=sampler_name,
batch_size=1,
n_iter=1,
steps=p.steps,
steps=steps,
cfg_scale=cfg_scale,
width=width,
height=height,