Skip to content

Commit 8ec076f

Browse files
Isotr0pygarg-amit
authored andcommitted
[CI/Build] Enable InternVL2 PP test only on single node (vllm-project#8437)
Signed-off-by: Amit Garg <mitgarg17495@gmail.com>
1 parent fb8cd6e commit 8ec076f

File tree

1 file changed

+5
-4
lines changed

1 file changed

+5
-4
lines changed

tests/distributed/test_pipeline_parallel.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -32,10 +32,11 @@
3232
(1, 4, 1, 0, 0, "meta-llama/Meta-Llama-3-8B", "ray"),
3333
(2, 2, 1, 0, 0, "meta-llama/Meta-Llama-3-8B", "ray"),
3434
(2, 2, 0, 1, 0, "meta-llama/Meta-Llama-3-8B", "ray"),
35-
# TODO: Enable internVL2 in a separate test if needed
36-
# (1, 2, 1, 1, 1, "OpenGVLab/InternVL2-1B", "ray"),
37-
# (1, 2, 1, 1, 1, "OpenGVLab/InternVL2-2B", "ray"),
38-
# (1, 2, 1, 0, 1, "OpenGVLab/InternVL2-4B", "ray"),
35+
# NOTE: InternVL2 multi-node tests are flaky,
36+
# use mp backend to skip the multi-node tests
37+
(1, 2, 1, 1, 1, "OpenGVLab/InternVL2-1B", "mp"),
38+
(1, 2, 1, 1, 1, "OpenGVLab/InternVL2-2B", "mp"),
39+
(1, 2, 1, 0, 1, "OpenGVLab/InternVL2-4B", "mp"),
3940
],
4041
)
4142
@fork_new_process_for_each_test

0 commit comments

Comments
 (0)