Flax schedulers: replace control flow with functions#577

pcuenca wants to merge 1 commit intomainhuggingface/diffusers:mainfrom

functional-schedulershuggingface/diffusers:functional-schedulersCopy head branch name to clipboard