Diffusion model toolset