el_output (torch.FloatTensor) — direct output from learned diffusion model.


sigma_hat (float) — TODO


sigma_prev (float) — TODO


sample_hat (torch.FloatTensor) — TODO


return_dict (bool) — option for returning tuple rather than KarrasVeOutput class
KarrasVeOutput — updated sample in the diffusion chain and derivative (TODO double check).


Returns

KarrasVeOutput or tuple



KarrasVeOutput if return_dict is True, otherwise a tuple. When
re