Diffusion Model In Python How To Comment Out Code