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: separate checkpoint

Browse Source
This commit is contained in:
Bingsu
2023-08-31 20:35:37 +09:00
parent 9bb559a0b1
commit 9f83506c35
4 changed files with 55 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
scripts/!adetailer.py
View File

@@ -125,6 +125,7 @@ class AfterDetailerScript(scripts.Script):
sampler_names=sampler_names,
t2i_button=txt2img_submit_button,
i2i_button=img2img_submit_button,
checkpoints_list=modules.sd_models.checkpoint_tiles,
)
components, infotext_fields = adui(num_models, is_img2img, webui_info)
@@ -307,8 +308,16 @@ class AfterDetailerScript(scripts.Script):
def get_override_settings(self, p, args: ADetailerArgs) -> dict[str, Any]:
d = {}
if args.ad_use_clip_skip:
d["CLIP_stop_at_last_layers"] = args.ad_clip_skip
if (
args.ad_use_checkpoint
and args.ad_checkpoint
and args.ad_checkpoint not in ("None", "Use same checkpoint")
):
d["sd_model_checkpoint"] = args.ad_checkpoint
return d
def get_initial_noise_multiplier(self, p, args: ADetailerArgs) -> float | None: