diff --git a/.buildkite/intel_jobs/models_multimodal_intel.yaml b/.buildkite/intel_jobs/models_multimodal_intel.yaml index 0e12690604..42d429f007 100644 --- a/.buildkite/intel_jobs/models_multimodal_intel.yaml +++ b/.buildkite/intel_jobs/models_multimodal_intel.yaml @@ -125,7 +125,5 @@ steps: pip install open-clip-torch --no-deps && cd tests && pytest -v -s models/multimodal/processing/test_tensor_schema.py - --deselect "tests/models/multimodal/processing/test_tensor_schema.py::test_model_tensor_schema[mistralai/Mistral-Large-3-675B-Instruct-2512-NVFP4]" - --deselect "tests/models/multimodal/processing/test_tensor_schema.py::test_model_tensor_schema[Qwen/Qwen2.5-Omni-7B-AWQ]" --num-shards=$$BUILDKITE_PARALLEL_JOB_COUNT --shard-id=$$BUILDKITE_PARALLEL_JOB' parallelism: 4 diff --git a/tests/models/multimodal/processing/test_common.py b/tests/models/multimodal/processing/test_common.py index ea5aeb8e2c..f785a68f97 100644 --- a/tests/models/multimodal/processing/test_common.py +++ b/tests/models/multimodal/processing/test_common.py @@ -20,6 +20,7 @@ from vllm.multimodal import MULTIMODAL_REGISTRY from vllm.multimodal.cache import MultiModalProcessorOnlyCache from vllm.multimodal.inputs import batched_tensors_equal from vllm.multimodal.processing import BaseMultiModalProcessor, InputProcessingContext +from vllm.platforms import current_platform from vllm.tokenizers import TokenizerLike, cached_tokenizer_from_config from vllm.utils.mistral import is_mistral_tokenizer @@ -83,6 +84,12 @@ MM_DATA_PATCHES = { "glmasr": glmasr_patch_mm_data, } +_XPU_EXCLUDED_MODEL_IDS = { + "baidu/Unlimited-OCR", + "mistralai/Mistral-Large-3-675B-Instruct-2512-NVFP4", + "Qwen/Qwen2.5-Omni-7B-AWQ", +} + def _iter_model_ids_to_test(model_arch_list: AbstractSet[str]): for model_arch in model_arch_list: @@ -97,7 +104,14 @@ def _iter_model_ids_to_test(model_arch_list: AbstractSet[str]): def _get_model_ids_to_test(model_arch_list: AbstractSet[str]): - return list(_iter_model_ids_to_test(model_arch_list)) + model_ids = list(_iter_model_ids_to_test(model_arch_list)) + + if current_platform.is_xpu(): + for excluded_model_id in _XPU_EXCLUDED_MODEL_IDS: + while excluded_model_id in model_ids: + model_ids.remove(excluded_model_id) + + return model_ids def get_model_ids_to_test():