std1.0.0-beta-7 SafeC docs

std :: ml::diffusion

Function std::dpm_solver_pp_2_step

#include <std/ml/diffusion.h> source

generic<M: DenoiseModel> &FloatTensor dpm_solver_pp_2_step(
    &M model,
    const &FloatTensor x_cur,
    const &FloatTensor x0Pred,
    float alpha_cur,
    float sigma_cur,
    float alpha_next,
    float sigma_next,
)

Order-2 midpoint variant, x0-prediction form: model.predict(x_s, sigma_s) at the log-SNR midpoint returns a predicted x0 (not eps).