|
32 | 32 | (1, 4, 1, 0, 0, "meta-llama/Meta-Llama-3-8B", "ray"),
|
33 | 33 | (2, 2, 1, 0, 0, "meta-llama/Meta-Llama-3-8B", "ray"),
|
34 | 34 | (2, 2, 0, 1, 0, "meta-llama/Meta-Llama-3-8B", "ray"),
|
35 |
| - # TODO: Enable internVL2 in a separate test if needed |
36 |
| - # (1, 2, 1, 1, 1, "OpenGVLab/InternVL2-1B", "ray"), |
37 |
| - # (1, 2, 1, 1, 1, "OpenGVLab/InternVL2-2B", "ray"), |
38 |
| - # (1, 2, 1, 0, 1, "OpenGVLab/InternVL2-4B", "ray"), |
| 35 | + # NOTE: InternVL2 multi-node tests are flaky, |
| 36 | + # use mp backend to skip the multi-node tests |
| 37 | + (1, 2, 1, 1, 1, "OpenGVLab/InternVL2-1B", "mp"), |
| 38 | + (1, 2, 1, 1, 1, "OpenGVLab/InternVL2-2B", "mp"), |
| 39 | + (1, 2, 1, 0, 1, "OpenGVLab/InternVL2-4B", "mp"), |
39 | 40 | ],
|
40 | 41 | )
|
41 | 42 | @fork_new_process_for_each_test
|
|
0 commit comments