Yes, should be the same as the equation before. Like this: xts = alpha_bar[t].sqrt() * x0s + (1.-alpha_bar[t]).sqrt() * eps Additionally, the code isn't con

Yes, should be the same as the equation before. Like this: xts = alpha_bar[t].sqrt() * x0s + (1.-alpha_bar[t]).sqrt() * eps Additionally, the code isn't consistent. In the sampling code a time embedding is used, while in training it isn't.