From f425968b0258964aede4248491b81c3f0658c8ca Mon Sep 17 00:00:00 2001
From: Roger Wang <136131678+ywang96@users.noreply.github.com>
Date: Thu, 12 Sep 2024 15:05:35 -0700
Subject: [PATCH] [CI/Build] Disable multi-node test for InternVL2 (#8428)

---
 tests/distributed/test_pipeline_parallel.py | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/tests/distributed/test_pipeline_parallel.py b/tests/distributed/test_pipeline_parallel.py
index d2219eed988e1..9a02f468f0a93 100644
--- a/tests/distributed/test_pipeline_parallel.py
+++ b/tests/distributed/test_pipeline_parallel.py
@@ -32,9 +32,10 @@
         (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