riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-16 19:18:24 +06:00
Code Issues Actions 38 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
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 1 deletions
Show Stats 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",