From d8df3b8111d0e94db6eb7d9674e63424b7e84125 Mon Sep 17 00:00:00 2001 From: Dowon Date: Sun, 14 Apr 2024 09:18:57 +0900 Subject: [PATCH] fix(scripts): scheduler import error --- adetailer/ui.py | 4 ++-- scripts/!adetailer.py | 6 +++--- 2 files changed, 5 insertions(+), 5 deletions(-) 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)