Skip to content

GitLab

Unverified Commit cf8cc326 authored by Huy Do's avatar Huy Do Committed by GitHub
Browse files

Fix a couple of Voxtral tests (#21218) 


Signed-off-by: default avatarHuy Do <huydhn@gmail.com>
parent 3a2cb264
Hide whitespace changes
Inline Side-by-side
Showing with 5 additions and 1 deletion
tests/models/registry.py
View file @ cf8cc326
...@@ -449,7 +449,11 @@ _MULTIMODAL_EXAMPLE_MODELS = { ...@@ -449,7 +449,11 @@ _MULTIMODAL_EXAMPLE_MODELS = {
tokenizer="Isotr0py/Florence-2-tokenizer", # noqa: E501 tokenizer="Isotr0py/Florence-2-tokenizer", # noqa: E501
trust_remote_code=True), # noqa: E501 trust_remote_code=True), # noqa: E501
"MllamaForConditionalGeneration": _HfExamplesInfo("meta-llama/Llama-3.2-11B-Vision-Instruct"), # 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 "WhisperForConditionalGeneration": _HfExamplesInfo("openai/whisper-large-v3"), # noqa: E501
# [Cross-encoder] # [Cross-encoder]
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment