cmp.stages.diffusion.diffusion module