transformers

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

#32050

Merged