Diffusion Model Tensorflow Js