Conditional Diffusion Model Pytorch