Diffusion Model In Python Example Code Guilhem Castagne