fix(model): restore Qwen3-Omni audio with Transformers 5.14

Signed-off-by: bvolpato <brunocvcunha@gmail.com>

Author: bvolpato

src/megatron/bridge/models/qwen_omni/modeling_qwen3_omni/thinker_model.py

@@ -28,7 +28,10 @@
     Qwen3OmniMoeVisionEncoder as Qwen3OmniMoeVisionEncoderHF,
 )
 
-from megatron.bridge.models.qwen_omni.modeling_qwen3_omni.rope import get_rope_index
+from megatron.bridge.models.qwen_omni.modeling_qwen3_omni.rope import (
+    _get_feat_extract_output_lengths,
+    get_rope_index,
+)
 from megatron.bridge.models.qwen_omni.modeling_qwen3_omni.transformer_config import (
     Qwen3OmniTransformerConfig,
 )
@@ -337,7 +340,7 @@ def get_audio_features(
         if expected_audio_token_counts is None:
             return audio_embeds
 
-        _, produced_output_lengths = self.audio_model._get_feat_extract_output_lengths(audio_feature_lengths)
+        produced_output_lengths = _get_feat_extract_output_lengths(audio_feature_lengths)
         expected_audio_token_counts = expected_audio_token_counts.to(produced_output_lengths.device)
 
         trimmed_embeds = []