Bing-su/adetailer
1
0
Fork 0
mirror of https://github.com/Bing-su/adetailer.git synced 2026-03-05 21:40:02 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: function -> list

Browse Source
This commit is contained in:
Bingsu
2023-09-10 22:07:02 +09:00
parent 64ea03e571
commit 645c67e3ff
2 changed files with 13 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
scripts/!adetailer.py
View File

@@ -120,13 +120,20 @@ class AfterDetailerScript(scripts.Script):
num_models = opts.data.get("ad_max_models", 2)
ad_model_list = list(model_mapping.keys())
sampler_names = [sampler.name for sampler in all_samplers]
try:
checkpoint_list = modules.sd_models.checkpoint_tiles(use_shorts=True)
except TypeError:
checkpoint_list = modules.sd_models.checkpoint_tiles()
vae_list = modules.shared_items.sd_vae_items()
webui_info = WebuiInfo(
ad_model_list=ad_model_list,
sampler_names=sampler_names,
t2i_button=txt2img_submit_button,
i2i_button=img2img_submit_button,
checkpoints_list=modules.sd_models.checkpoint_tiles,
vae_list=modules.shared_items.sd_vae_items,
checkpoints_list=checkpoint_list,
vae_list=vae_list,
)
components, infotext_fields = adui(num_models, is_img2img, webui_info)