Converts a denoiser output to a Karras ODE derivative.
(x, sigma, denoised)
| 44 | |
| 45 | |
| 46 | def to_d(x, sigma, denoised): |
| 47 | """Converts a denoiser output to a Karras ODE derivative.""" |
| 48 | return (x - denoised) / utils.append_dims(sigma, x.ndim) |
| 49 | |
| 50 | |
| 51 | def get_ancestral_step(sigma_from, sigma_to, eta=1.): |
no outgoing calls
no test coverage detected