TEACHING DIFFUSION MODELS TO COUNT

Summary from the TLDR AI newsletter

Getting a diffusion model to generate a certain number of an object is currently challenging. This work introduces a counting token that allows a model to successfully generate just a few or very many of a certain object. It isn't perfect as it builds on base stable diffusion, but it does much better than current alternatives.