Small model updates  [skip ci]