AUTOMATIC1111 / stable-diffusion-webui

Stable Diffusion web UI
GNU Affero General Public License v3.0
131.68k stars 25.27k forks source link

[Feature Request]: Implement AYS Sampler proposed by Nvidia #15669

Open GongXinyuu opened 2 weeks ago

GongXinyuu commented 2 weeks ago

Is there an existing issue for this?

What would your feature do ?

Align Your Steps (AYS), a novel general framework for optimizing sampling schedules in diffusion models that significantly boosts the quality of outputs, especially when performing synthesis in few steps.

image

Proposed workflow

The author has implemented AYS sampler in diffusers. It would be convenient to port it into webui. https://research.nvidia.com/labs/toronto-ai/AlignYourSteps/howto.html

Additional information

  1. https://arxiv.org/abs/2404.14507
  2. https://research.nvidia.com/labs/toronto-ai/AlignYourSteps/