From cf8cc32674f30cc091b551ceb4decd79718ac9e5 Mon Sep 17 00:00:00 2001 From: Huy Do Date: Sat, 19 Jul 2025 02:13:41 -0700 Subject: [PATCH] Fix a couple of Voxtral tests (#21218) Signed-off-by: Huy Do --- tests/models/registry.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/tests/models/registry.py b/tests/models/registry.py index 095e6f590119..5c546a6c86da 100644 --- a/tests/models/registry.py +++ b/tests/models/registry.py @@ -449,7 +449,11 @@ _MULTIMODAL_EXAMPLE_MODELS = { tokenizer="Isotr0py/Florence-2-tokenizer", # noqa: E501 trust_remote_code=True), # noqa: E501 "MllamaForConditionalGeneration": _HfExamplesInfo("meta-llama/Llama-3.2-11B-Vision-Instruct"), # noqa: E501 - "VoxtralForConditionalGeneration": _HfExamplesInfo("mistralai/Voxtral-Mini-3B-2507", tokenizer_mode="mistral"), # noqa: E501 + "VoxtralForConditionalGeneration": _HfExamplesInfo( + "mistralai/Voxtral-Mini-3B-2507", + tokenizer_mode="mistral", + min_transformers_version="4.54" + ), "WhisperForConditionalGeneration": _HfExamplesInfo("openai/whisper-large-v3"), # noqa: E501 # [Cross-encoder]