:red_circle: :red_circle: fix `query_pre_attn_scalar` different of `num_heads` in default gemma2 config (#34540)
* fix query_pre_attn_scalar different of num_heads in default config
* propagate modular changes
* fix copies
* fix modular copies
* fix copies?
* correct copies fix