Comment by tzvc

Lots of tuning to get the model to not immediatly spiral into nonsense. But small models are getting better by the minute, maybe i'll revisit it with a better model and share all the code