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

b61bd98f · Roger Wang · GitHub · c1636945 · b61bd98f
Unverified Commit b61bd98f authored Sep 12, 2024 by Roger Wang Committed by GitHub Sep 12, 2024
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 3 deletions

tests/distributed/test_pipeline_parallel.py tests/distributed/test_pipeline_parallel.py +4 -3

No files found.
--- a/tests/distributed/test_pipeline_parallel.py
+++ b/tests/distributed/test_pipeline_parallel.py
@@ -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"),
        (2, 2, 1, 0, 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"),
-        (1, 2, 1, 1, 1, "OpenGVLab/InternVL2-2B", "ray"),
-        (1, 2, 1, 0, 1, "OpenGVLab/InternVL2-4B", "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"),
    ],
 )
 @fork_new_process_for_each_test