1580158
/
adetailer
mirror of https://github.com/Bing-su/adetailer.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix(scripts): fix optioninfo, add xyz options

pull/728/head
Dowon 2024-09-17 19:42:42 +09:00
parent ceacc711e4
commit feefeed638
No known key found for this signature in database
GPG Key ID: 31D3079F422E1444
1 changed files with 29 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

35
scripts/!adetailer.py
View File

@ -977,18 +977,22 @@ def on_ui_settings():
shared.opts.add_option( shared.opts.add_option(
"ad_save_previews", "ad_save_previews",
shared.OptionInfo(False, "Save mask previews", section=section), shared.OptionInfo(default=False, label="Save mask previews", section=section),
) )
shared.opts.add_option( shared.opts.add_option(
"ad_save_images_before", "ad_save_images_before",
shared.OptionInfo(False, "Save images before ADetailer", section=section), shared.OptionInfo(
default=False, label="Save images before ADetailer", section=section
),
) )
shared.opts.add_option( shared.opts.add_option(
"ad_only_selected_scripts", "ad_only_selected_scripts",
shared.OptionInfo( shared.OptionInfo(
True, "Apply only selected scripts to ADetailer", section=section default=True,
label="Apply only selected scripts to ADetailer",
section=section,
), ),
) )
@ -1022,7 +1026,9 @@ def on_ui_settings():
shared.opts.add_option( shared.opts.add_option(
"ad_same_seed_for_each_tab", "ad_same_seed_for_each_tab",
shared.OptionInfo( shared.OptionInfo(
False, "Use same seed for each tab in adetailer", section=section default=False,
label="Use same seed for each tab in adetailer",
section=section,
), ),
) )
@ -1088,7 +1094,8 @@ def make_axis_on_xyz_grid():
return return
model_list = ["None", *model_mapping.keys()] model_list = ["None", *model_mapping.keys()]
samplers = [sampler.name for sampler in all_samplers] xyz_samplers = [sampler.name for sampler in all_samplers]
xyz_schedulers = [scheduler.label for scheduler in schedulers]
axis = [ axis = [
xyz_grid.AxisOption( xyz_grid.AxisOption(
@ -1127,6 +1134,11 @@ def make_axis_on_xyz_grid():
float, float,
partial(set_value, field="ad_denoising_strength"), partial(set_value, field="ad_denoising_strength"),
), ),
xyz_grid.AxisOption(
"[ADetailer] CFG scale 1st",
float,
partial(set_value, field="ad_cfg_scale"),
),
xyz_grid.AxisOption( xyz_grid.AxisOption(
"[ADetailer] Inpaint only masked 1st", "[ADetailer] Inpaint only masked 1st",
str, str,
@ -1142,7 +1154,18 @@ def make_axis_on_xyz_grid():
"[ADetailer] ADetailer sampler 1st", "[ADetailer] ADetailer sampler 1st",
str, str,
partial(set_value, field="ad_sampler"), partial(set_value, field="ad_sampler"),
choices=lambda: samplers, choices=lambda: xyz_samplers,
),
xyz_grid.AxisOption(
"[ADetailer] ADetailer scheduler 1st",
str,
partial(set_value, field="ad_scheduler"),
choices=lambda: xyz_schedulers,
),
xyz_grid.AxisOption(
"[ADetailer] noise multiplier 1st",
float,
partial(set_value, field="ad_noise_multiplier"),
), ),
xyz_grid.AxisOption( xyz_grid.AxisOption(
"[ADetailer] ControlNet model 1st", "[ADetailer] ControlNet model 1st",