Diffusion Model Class