Implement: Align Your Steps: Optimizing Sampling Schedules in Diffusion Models

Align Your Steps: Optimizing Sampling Schedules in Diffusion Models is a general and principled approach to optimizing the sampling schedules of DMs for high-quality outputs.
This work is presented by Nvidia labs in this paper:
https://arxiv.org/abs/2404.14507

and the project page is here
https://research.nvidia.com/labs/toronto-ai/AlignYourSteps/

the page propose a very small change that have big impact on the inference quality