Fix Qwen3OmniMoE weight init (#42531)

* module.router -> module.gate

* i am not smart today

Author: Rocketknight1

src/transformers/models/qwen3_omni_moe/modeling_qwen3_omni_moe.py

@@ -84,7 +84,7 @@ def _init_weights(self, module):
         if isinstance(module, Qwen3OmniMoeThinkerTextSparseMoeBlock):
             init.normal_(module.experts.gate_up_proj, mean=0.0, std=std)
             init.normal_(module.experts.down_proj, mean=0.0, std=std)
-            init.normal_(module.router.weight, mean=0.0, std=std)
+            init.normal_(module.gate.weight, mean=0.0, std=std)
 
 
 def _get_feat_extract_output_lengths(input_lengths):

src/transformers/models/qwen3_omni_moe/modular_qwen3_omni_moe.py

@@ -899,7 +899,7 @@ def _init_weights(self, module):
         if isinstance(module, Qwen3OmniMoeThinkerTextSparseMoeBlock):
             init.normal_(module.experts.gate_up_proj, mean=0.0, std=std)
             init.normal_(module.experts.down_proj, mean=0.0, std=std)
-            init.normal_(module.router.weight, mean=0.0, std=std)
+            init.normal_(module.gate.weight, mean=0.0, std=std)
 
 
 class Qwen3OmniMoePreTrainedModelForConditionalGeneration(Qwen2_5OmniPreTrainedModelForConditionalGeneration):