I was thinking that this would be low-hanging fruit for a model. The parameter space is so tiny compared to what a diffusion model already has to deal with...