[tests, qwen2_5_omni] fix flaky tests (#37721)

Author: gante

tests/generation/test_utils.py

@@ -134,7 +134,7 @@
     "mistral3",
     "chameleon",
     "internvl",
-    "qwen2_5_omni",
+    "qwen2_5omni",  # the file is named `qwen2_5_omni`, but the model class is `Qwen2_5Omni`
 ]
 
 
@@ -230,6 +230,7 @@ def _get_logits_processor_kwargs(self, do_sample=False, config=None):
             for key in [
                 "image_token_id",
                 "video_token_id",
+                "audio_token_id",
                 "vision_start_token_id",
                 "audio_start_token_id",
                 "audio_end_token_id",