A Text2Text model for semantic generation of building layouts

Sure! I have loads actually :slight_smile:

These are some outputs from training the smallest neo (125M) on 5% of my total data.

Legend: bedroom=green, bathroom=pink, the other two are kitchen (darker pink) and living room

My goal is to understand the effect of scale in this, going all the way up to Neo-J 6b model trained on all the data.

But also, ideally, to incorporate more modalities such as taking advantage of relational data, adding design evaluation as rewards, and other things I can’t think of but others might!

3 Likes