update diffusion model interface