Diffusion Model In Python How To Comment