Skip to content

Commit 699b21c

Browse files
DarkLight1337ZhengHongming888
authored andcommitted
Revert "[CI/Build] Use CPU for mm processing test on CI (vllm-project#27522)" (vllm-project#27531)
Signed-off-by: DarkLight1337 <[email protected]>
1 parent 32663f4 commit 699b21c

File tree

1 file changed

+2
-4
lines changed

1 file changed

+2
-4
lines changed

.buildkite/test-pipeline.yaml

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

736-
- label: Multi-Modal Processor and Models Test (CPU) # 44min
736+
- label: Multi-Modal Processor Test # 44min
737737
timeout_in_minutes: 60
738-
no_gpu: true
739738
source_file_dependencies:
740739
- vllm/
741740
- tests/models/multimodal
742741
commands:
743742
- pip install git+https://github.com/TIGER-AI-Lab/Mantis.git
744743
- pytest -v -s models/multimodal/processing
745-
- pytest -v -s models/multimodal/test_mapping.py
746744

747745
- label: Multi-Modal Models Test (Standard) # 60min
748746
timeout_in_minutes: 80
@@ -754,7 +752,7 @@ steps:
754752
commands:
755753
- pip install git+https://github.com/TIGER-AI-Lab/Mantis.git
756754
- pip freeze | grep -E 'torch'
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
755+
- pytest -v -s models/multimodal -m core_model --ignore models/multimodal/generation/test_whisper.py --ignore models/multimodal/processing
758756
- 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
759757

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

0 commit comments

Comments
 (0)