Diffusion model - draft | ScratchStats