Yeah. I wasn't really sure what to do here, since models also seem to know about inference (?)