replace diffusionInfo with diffusion_info