Diffusion Model Tutorial Pytorch Lightning