Diffusion Model Tensorflow Tutorial Python