Diffusion Pytorch Mode