Skip to content

GitLab

Unverified Commit cef65f07 authored by Andreas Karatzas's avatar Andreas Karatzas Committed by GitHub
Browse files

[ROCm][CI] Removed hard-coded attn backend requirement for Qwen VL (#34753) 


Signed-off-by: default avatarAndreas Karatzas <akaratza@amd.com>
parent 6f3b2047
Show whitespace changes
Inline Side-by-side
Showing with 0 additions and 7 deletions
tests/models/multimodal/generation/test_common.py
View file @ cef65f07
...@@ -169,13 +169,6 @@ VLM_TEST_SETTINGS = { ...@@ -169,13 +169,6 @@ VLM_TEST_SETTINGS = {
auto_cls=AutoModelForImageTextToText, auto_cls=AutoModelForImageTextToText,
vllm_output_post_proc=model_utils.qwen2_vllm_to_hf_output, vllm_output_post_proc=model_utils.qwen2_vllm_to_hf_output,
patch_hf_runner=model_utils.qwen3_vl_patch_hf_runner, patch_hf_runner=model_utils.qwen3_vl_patch_hf_runner,
vllm_runner_kwargs={
"attention_config": {
"backend": "ROCM_AITER_FA",
},
}
if current_platform.is_rocm()
else None,
image_size_factors=[(0.25,), (0.25, 0.25, 0.25), (0.25, 0.2, 0.15)], image_size_factors=[(0.25,), (0.25, 0.25, 0.25), (0.25, 0.2, 0.15)],
marks=[ marks=[
pytest.mark.core_model, pytest.mark.core_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