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

fix(scripts): scheduler import error

Browse Source
This commit is contained in:
Dowon
2024-04-14 09:18:57 +09:00
parent 38e369305e
commit d8df3b8111
2 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
adetailer/ui.py
View File

@@ -553,8 +553,8 @@ def inpainting(w: Widgets, n: int, is_img2img: bool, webui_info: WebuiInfo):
] ]
w.ad_scheduler = gr.Dropdown( w.ad_scheduler = gr.Dropdown(
label="ADetailer scheduler" + suffix(n), label="ADetailer scheduler" + suffix(n),
choices=webui_info.scheduler_names, choices=scheduler_names,
value=webui_info.scheduler_names[0], value=scheduler_names[0],
visible=len(scheduler_names) > 2, visible=len(scheduler_names) > 2,
elem_id=eid("ad_scheduler"), elem_id=eid("ad_scheduler"),
) )

6
scripts/!adetailer.py
View File

@@ -76,11 +76,11 @@ except ImportError:
try: try:
from modules.sd_schedulers import sd_schedulers from modules.sd_schedulers import schedulers
scheduler_available = True scheduler_available = True
except ImportError: except ImportError:
sd_schedulers = [] schedulers = []
scheduler_available = False scheduler_available = False
@@ -127,7 +127,7 @@ class AfterDetailerScript(scripts.Script):
num_models = opts.data.get("ad_max_models", 2) num_models = opts.data.get("ad_max_models", 2)
ad_model_list = list(model_mapping.keys()) ad_model_list = list(model_mapping.keys())
sampler_names = [sampler.name for sampler in all_samplers] sampler_names = [sampler.name for sampler in all_samplers]
scheduler_names = [x.label for x in sd_schedulers] scheduler_names = [x.label for x in schedulers]
try: try:
checkpoint_list = modules.sd_models.checkpoint_tiles(use_shorts=True) checkpoint_list = modules.sd_models.checkpoint_tiles(use_shorts=True)