From 08ddf125219fb0740c10c01a63d13f0c22890f06 Mon Sep 17 00:00:00 2001 From: SLAPaper Date: Mon, 10 Jun 2024 02:29:57 +0800 Subject: [PATCH] feat: not add AYS when using latest dev webui --- README.md | 2 ++ scripts/ays.py | 4 ++++ 2 files changed, 6 insertions(+) diff --git a/README.md b/README.md index 794adba..46a4fea 100644 --- a/README.md +++ b/README.md @@ -27,3 +27,5 @@ Node code extracted from [smZNodes](https://github.com/shiimizu/ComfyUI_smZNodes The scheduler introduced by [NVIDIA](https://research.nvidia.com/labs/toronto-ai/AlignYourSteps/howto.html) **Recommended Sampler**: DPM++ 2M SDE + +Update: latest dev webui have already added these [scheduler](https://github.com/AUTOMATIC1111/stable-diffusion-webui/pull/15751) diff --git a/scripts/ays.py b/scripts/ays.py index 28892b5..13f1b33 100644 --- a/scripts/ays.py +++ b/scripts/ays.py @@ -90,6 +90,10 @@ def add_align_your_step_scheduler() -> None: except ImportError: return + # latest dev webui have already added these scheduler: https://github.com/AUTOMATIC1111/stable-diffusion-webui/pull/15751 + if 'align_your_steps' in sd_schedulers.schedulers_map: + return + scheduler_v15 = sd_schedulers.Scheduler( "ays_v15", "Align Your Step SD15", align_your_step_scheduler_v15 )