Skip to content

Commit 5672c2b

Browse files
charlotte12lskyloevil
authored andcommitted
[CI]Add transformers_utils to Async Engine, Inputs, Utils, Worker Test (vllm-project#24615)
Signed-off-by: Xingyu Liu <charlotteliu12x@gmail.com>
1 parent 67a62ef commit 5672c2b

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
@@ -54,6 +54,7 @@ steps:
5454
- tests/utils_
5555
- tests/worker
5656
- tests/standalone_tests/lazy_imports.py
57+
- tests/transformers_utils
5758
commands:
5859
- python3 standalone_tests/lazy_imports.py
5960
- pytest -v -s mq_llm_engine # MQLLMEngine
@@ -63,6 +64,7 @@ steps:
6364
- pytest -v -s multimodal
6465
- pytest -v -s utils_ # Utils
6566
- pytest -v -s worker # Worker
67+
- pytest -v -s transformers_utils # transformers_utils
6668

6769
- label: Python-only Installation Test # 10min
6870
timeout_in_minutes: 20
@@ -822,8 +824,8 @@ steps:
822824
# Avoid importing model tests that cause CUDA reinitialization error
823825
- pytest models/test_transformers.py -v -s -m 'distributed(num_gpus=2)'
824826
- pytest models/language -v -s -m 'distributed(num_gpus=2)'
825-
- pytest models/multimodal -v -s -m 'distributed(num_gpus=2)' --ignore models/multimodal/generation/test_whisper.py
826-
- VLLM_WORKER_MULTIPROC_METHOD=spawn pytest models/multimodal/generation/test_whisper.py -v -s -m 'distributed(num_gpus=2)'
827+
- pytest models/multimodal -v -s -m 'distributed(num_gpus=2)' --ignore models/multimodal/generation/test_whisper.py
828+
- VLLM_WORKER_MULTIPROC_METHOD=spawn pytest models/multimodal/generation/test_whisper.py -v -s -m 'distributed(num_gpus=2)'
827829
# test sequence parallel
828830
- pytest -v -s distributed/test_sequence_parallel.py
829831
# this test fails consistently.

0 commit comments

Comments
 (0)