Update to use the latest kiji-modeling