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.