From d63cd9ff1000a75cc88564e78317fd31d988dfa9 Mon Sep 17 00:00:00 2001 From: Isotr0py Date: Sun, 26 Oct 2025 13:09:18 +0800 Subject: [PATCH] [CI/Build] Use CPU for mm processing test on CI (#27522) Signed-off-by: Isotr0py --- .buildkite/test-pipeline.yaml | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/.buildkite/test-pipeline.yaml b/.buildkite/test-pipeline.yaml index 3f1d50d558109..0a7a871fb9c5c 100644 --- a/.buildkite/test-pipeline.yaml +++ b/.buildkite/test-pipeline.yaml @@ -733,14 +733,16 @@ steps: commands: - pytest -v -s models/language/pooling_mteb_test -- label: Multi-Modal Processor Test # 44min +- label: Multi-Modal Processor and Models Test (CPU) # 44min timeout_in_minutes: 60 + no_gpu: true source_file_dependencies: - vllm/ - tests/models/multimodal commands: - pip install git+https://github.com/TIGER-AI-Lab/Mantis.git - pytest -v -s models/multimodal/processing + - pytest -v -s models/multimodal/test_mapping.py - label: Multi-Modal Models Test (Standard) # 60min timeout_in_minutes: 80 @@ -752,7 +754,7 @@ steps: commands: - pip install git+https://github.com/TIGER-AI-Lab/Mantis.git - pip freeze | grep -E 'torch' - - pytest -v -s models/multimodal -m core_model --ignore models/multimodal/generation/test_whisper.py --ignore models/multimodal/processing + - pytest -v -s models/multimodal -m core_model --ignore models/multimodal/test_mapping.py --ignore models/multimodal/generation/test_whisper.py --ignore models/multimodal/processing - cd .. && VLLM_WORKER_MULTIPROC_METHOD=spawn pytest -v -s tests/models/multimodal/generation/test_whisper.py -m core_model # Otherwise, mp_method="spawn" doesn't work - label: Multi-Modal Accuracy Eval (Small Models) # 50min