Diffusion Model In Python Mean