Skip to content

Commit 8a1492d

Browse files
Isotr0pyrtourgeman
authored andcommitted
[CI/Build] Use CPU for mm processing test on CI (vllm-project#27522)
Signed-off-by: Isotr0py <mozf@mail2.sysu.edu.cn>
1 parent 5cda09f commit 8a1492d

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

.buildkite/test-pipeline.yaml

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -733,14 +733,16 @@ steps:
733733
commands:
734734
- pytest -v -s models/language/pooling_mteb_test
735735

736-
- label: Multi-Modal Processor Test # 44min
736+
- label: Multi-Modal Processor and Models Test (CPU) # 44min
737737
timeout_in_minutes: 60
738+
no_gpu: true
738739
source_file_dependencies:
739740
- vllm/
740741
- tests/models/multimodal
741742
commands:
742743
- pip install git+https://github.com/TIGER-AI-Lab/Mantis.git
743744
- pytest -v -s models/multimodal/processing
745+
- pytest -v -s models/multimodal/test_mapping.py
744746

745747
- label: Multi-Modal Models Test (Standard) # 60min
746748
timeout_in_minutes: 80
@@ -752,7 +754,7 @@ steps:
752754
commands:
753755
- pip install git+https://github.com/TIGER-AI-Lab/Mantis.git
754756
- pip freeze | grep -E 'torch'
755-
- pytest -v -s models/multimodal -m core_model --ignore models/multimodal/generation/test_whisper.py --ignore models/multimodal/processing
757+
- 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
756758
- 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
757759

758760
- label: Multi-Modal Accuracy Eval (Small Models) # 50min

0 commit comments

Comments
 (0)