A little bit better generative model

with manual infinity but still not reasonable