Function std::ddim_sampler_step

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

&FloatTensor ddim_sampler_step(
    const &FloatTensor x_t,
    const &FloatTensor epsPred,
    float alphaBar_t,
    float alphaBar_prev,
)

x0Pred = (x_t - sqrt(1-alphaBar_t)*epsPred) / sqrt(alphaBar_t) x_{t-1} = sqrt(alphaBar_prev)*x0Pred + sqrt(1-alphaBar_prev)*epsPred