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 )