The ODE in this paper (arxiv.org/abs/2206.00364) lets black box ODE solvers work well with diffusion model ODEs! I tried torchdiffeq's adaptive_heun method and got this result with 38 fevals:

Jun 16, 2022 · 2:53 PM UTC

6
8
79
It's from my cc12m_1_cfg model, and the prompt was "New York City, oil on canvas".
8
Replying to @RiversHaveWings
Congratulations! Amazing results. And thanks for the paper recommendation.
1
Replying to @RiversHaveWings
To me, what is more amazing is how they pull an ODE as a substitute for an SDE. That’s a neat trick, justified by the (completely valid) “because it works” argument.
1
Replying to @RiversHaveWings
Look like what I get using JAX
3
Replying to @RiversHaveWings
Woah. This looks super effective.
1
Replying to @RiversHaveWings
2