Diffusion Model In Python Example