Diffusion Model In Python Example Code