cmp.stages.diffusion.reconstruction module