Converts a denoiser output to a Karras ODE derivative.
(x, sigma, denoised)
| 49 | |
| 50 | |
| 51 | def to_d(x, sigma, denoised): |
| 52 | """Converts a denoiser output to a Karras ODE derivative.""" |
| 53 | return (x - denoised) / append_dims(sigma, x.ndim) |
| 54 | |
| 55 | |
| 56 | def get_ancestral_step(sigma_from, sigma_to, eta=1.): |
no test coverage detected