Skip to content

GitLab

Unverified Commit 9b4a3b23 authored by Isotr0py's avatar Isotr0py Committed by GitHub
Browse files

[CI/Build] Enable InternVL2 PP test only on single node (#8437)

parent acda0b35
Show whitespace changes
Inline Side-by-side
Showing with 5 additions and 4 deletions
tests/distributed/test_pipeline_parallel.py
View file @ 9b4a3b23
......@@ -32,10 +32,11 @@ VLLM_MULTI_NODE = os.getenv("VLLM_MULTI_NODE", "0") == "1"
(1, 4, 1, 0, 0, "meta-llama/Meta-Llama-3-8B", "ray"),
(2, 2, 1, 0, 0, "meta-llama/Meta-Llama-3-8B", "ray"),
(2, 2, 0, 1, 0, "meta-llama/Meta-Llama-3-8B", "ray"),
# TODO: Enable internVL2 in a separate test if needed
# (1, 2, 1, 1, 1, "OpenGVLab/InternVL2-1B", "ray"),
# (1, 2, 1, 1, 1, "OpenGVLab/InternVL2-2B", "ray"),
# (1, 2, 1, 0, 1, "OpenGVLab/InternVL2-4B", "ray"),
# NOTE: InternVL2 multi-node tests are flaky,
# use mp backend to skip the multi-node tests
(1, 2, 1, 1, 1, "OpenGVLab/InternVL2-1B", "mp"),
(1, 2, 1, 1, 1, "OpenGVLab/InternVL2-2B", "mp"),
(1, 2, 1, 0, 1, "OpenGVLab/InternVL2-4B", "mp"),
],
)
@fork_new_process_for_each_test
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment