Diffusion Model Pytorch Tutorial