diff --git a/adetailer/ui.py b/adetailer/ui.py index a4f87fd..3b96a81 100644 --- a/adetailer/ui.py +++ b/adetailer/ui.py @@ -553,8 +553,8 @@ def inpainting(w: Widgets, n: int, is_img2img: bool, webui_info: WebuiInfo): ] w.ad_scheduler = gr.Dropdown( label="ADetailer scheduler" + suffix(n), - choices=webui_info.scheduler_names, - value=webui_info.scheduler_names[0], + choices=scheduler_names, + value=scheduler_names[0], visible=len(scheduler_names) > 2, elem_id=eid("ad_scheduler"), ) diff --git a/scripts/!adetailer.py b/scripts/!adetailer.py index 77ede1d..b24ad81 100644 --- a/scripts/!adetailer.py +++ b/scripts/!adetailer.py @@ -76,11 +76,11 @@ except ImportError: try: - from modules.sd_schedulers import sd_schedulers + from modules.sd_schedulers import schedulers scheduler_available = True except ImportError: - sd_schedulers = [] + schedulers = [] scheduler_available = False @@ -127,7 +127,7 @@ 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] - scheduler_names = [x.label for x in sd_schedulers] + scheduler_names = [x.label for x in schedulers] try: checkpoint_list = modules.sd_models.checkpoint_tiles(use_shorts=True)