Skip to content

GitLab

Unverified Commit b61bd98f authored by Roger Wang's avatar Roger Wang Committed by GitHub
Browse files

[CI/Build] Disable multi-node test for InternVL2 (#8428)

parent c1636945
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 3 deletions
tests/distributed/test_pipeline_parallel.py
View file @ b61bd98f
...@@ -32,9 +32,10 @@ VLLM_MULTI_NODE = os.getenv("VLLM_MULTI_NODE", "0") == "1" ...@@ -32,9 +32,10 @@ VLLM_MULTI_NODE = os.getenv("VLLM_MULTI_NODE", "0") == "1"
(1, 4, 1, 0, 0, "meta-llama/Meta-Llama-3-8B", "ray"), (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, 1, 0, 0, "meta-llama/Meta-Llama-3-8B", "ray"),
(2, 2, 0, 1, 0, "meta-llama/Meta-Llama-3-8B", "ray"), (2, 2, 0, 1, 0, "meta-llama/Meta-Llama-3-8B", "ray"),
(1, 2, 1, 1, 1, "OpenGVLab/InternVL2-1B", "ray"), # TODO: Enable internVL2 in a separate test if needed
(1, 2, 1, 1, 1, "OpenGVLab/InternVL2-2B", "ray"), # (1, 2, 1, 1, 1, "OpenGVLab/InternVL2-1B", "ray"),
(1, 2, 1, 0, 1, "OpenGVLab/InternVL2-4B", "ray"), # (1, 2, 1, 1, 1, "OpenGVLab/InternVL2-2B", "ray"),
# (1, 2, 1, 0, 1, "OpenGVLab/InternVL2-4B", "ray"),
], ],
) )
@fork_new_process_for_each_test @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