[CI/Build] Enable InternVL2 PP test only on single node (vllm-project…

…#8437)
KuntaiDu · Sep 13, 2024 · 09ad90e · 09ad90e
1 parent 4151d66
commit 09ad90e
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/tests/distributed/test_pipeline_parallel.py b/tests/distributed/test_pipeline_parallel.py
@@ -32,10 +32,11 @@
         (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