SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Joao Gante
2025-04-23 17:54:12 +01:00
committed by GitHub
parent 63c6331387
commit 80ea2c05c2
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
tests/generation/test_utils.py
View File

@@ -134,7 +134,7 @@ VLM_CLASS_NAMES = [
"mistral3", "mistral3",
"chameleon", "chameleon",
"internvl", "internvl",
"qwen2_5_omni", "qwen2_5omni", # the file is named `qwen2_5_omni`, but the model class is `Qwen2_5Omni`
] ]
@@ -230,6 +230,7 @@ class GenerationTesterMixin:
for key in [ for key in [
"image_token_id", "image_token_id",
"video_token_id", "video_token_id",
"audio_token_id",
"vision_start_token_id", "vision_start_token_id",
"audio_start_token_id", "audio_start_token_id",
"audio_end_token_id", "audio_end_token_id",