From e7c843b0890d580b5733a327ee35a4849e5fd3e6 Mon Sep 17 00:00:00 2001 From: Isotr0py <2037008807@qq.com> Date: Fri, 13 Sep 2024 14:35:20 +0800 Subject: [PATCH] [CI/Build] Enable InternVL2 PP test only on single node (#8437) --- tests/distributed/test_pipeline_parallel.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/tests/distributed/test_pipeline_parallel.py b/tests/distributed/test_pipeline_parallel.py index 9a02f468f0a93..02288dc9dac90 100644 --- 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