Trait std::DenoiseModel

trait DenoiseModel {
    &FloatTensor predict(const &FloatTensor x, float sigma);
}

A denoiser the order-2 solvers call at their midpoint: an eps-prediction model for dpm_solver_2_step, an x0-prediction one for dpm_solver_pp_2_step. Any struct with this method conforms.