Skip to content

GitLab

Unverified Commit bb8dab82 authored by Roger Wang's avatar Roger Wang Committed by GitHub
Browse files

[CI] Set max transformers version for Ultravox model test (#16149) 


Signed-off-by: default avatarRoger Wang <ywang@roblox.com>
parent fc0f8776
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
tests/models/registry.py
View file @ bb8dab82
...@@ -329,7 +329,8 @@ _MULTIMODAL_EXAMPLE_MODELS = { ...@@ -329,7 +329,8 @@ _MULTIMODAL_EXAMPLE_MODELS = {
min_transformers_version="4.49"), # noqa: E501 min_transformers_version="4.49"), # noqa: E501
"SkyworkR1VChatModel": _HfExamplesInfo("Skywork/Skywork-R1V-38B"), "SkyworkR1VChatModel": _HfExamplesInfo("Skywork/Skywork-R1V-38B"),
"UltravoxModel": _HfExamplesInfo("fixie-ai/ultravox-v0_5-llama-3_2-1b", # noqa: E501 "UltravoxModel": _HfExamplesInfo("fixie-ai/ultravox-v0_5-llama-3_2-1b", # noqa: E501
trust_remote_code=True), trust_remote_code=True,
max_transformers_version="4.50"),
# [Encoder-decoder] # [Encoder-decoder]
# Florence-2 uses BartFastTokenizer which can't be loaded from AutoTokenizer # Florence-2 uses BartFastTokenizer which can't be loaded from AutoTokenizer
# Therefore, we borrow the BartTokenizer from the original Bart model # Therefore, we borrow the BartTokenizer from the original Bart model
......
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