Diffusion Model Tensorflow Lite