Diffusion Model Tensorflow