Questions on the `BertModelLMHeadModel`

Yes it’s definitely a mistake. Will fix this morning.