[mistral] Support passing head_dim through config (and do not require head_dim * num_heads == hidden_size)#32050