Skip to content

GitLab

Unverified Commit d1d063a5 authored by Cyrus Leung's avatar Cyrus Leung Committed by GitHub
Browse files

[Chore] Use `max_transformers_version` for Qwen-VL test (#26792) 


Signed-off-by: default avatarDarkLight1337 <tlleungac@connect.ust.hk>
parent 7e6edb14
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
tests/models/multimodal/generation/test_common.py
View file @ d1d063a5
...@@ -707,8 +707,6 @@ VLM_TEST_SETTINGS = { ...@@ -707,8 +707,6 @@ VLM_TEST_SETTINGS = {
max_num_seqs=2, max_num_seqs=2,
vllm_output_post_proc=model_utils.qwen_vllm_to_hf_output, vllm_output_post_proc=model_utils.qwen_vllm_to_hf_output,
prompt_path_encoder=model_utils.qwen_prompt_path_encoder, prompt_path_encoder=model_utils.qwen_prompt_path_encoder,
# FIXME: https://github.com/huggingface/transformers/issues/38358
marks=[pytest.mark.skip("Model initialization fails")],
), ),
"qwen2_vl": VLMTestInfo( "qwen2_vl": VLMTestInfo(
models=["Qwen/Qwen2-VL-2B-Instruct"], models=["Qwen/Qwen2-VL-2B-Instruct"],
......
tests/models/registry.py
View file @ d1d063a5
...@@ -752,6 +752,8 @@ _MULTIMODAL_EXAMPLE_MODELS = { ...@@ -752,6 +752,8 @@ _MULTIMODAL_EXAMPLE_MODELS = {
"Qwen/Qwen-VL", "Qwen/Qwen-VL",
extras={"chat": "Qwen/Qwen-VL-Chat"}, extras={"chat": "Qwen/Qwen-VL-Chat"},
trust_remote_code=True, trust_remote_code=True,
max_transformers_version="4.53.3",
transformers_version_reason="Use of deprecated imports which have been removed.", # noqa: E501
hf_overrides={"architectures": ["QwenVLForConditionalGeneration"]}, hf_overrides={"architectures": ["QwenVLForConditionalGeneration"]},
), ),
"Qwen2AudioForConditionalGeneration": _HfExamplesInfo( "Qwen2AudioForConditionalGeneration": _HfExamplesInfo(
......
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