xinyun/vllm
1
0
Fork 0
mirror of https://git.datalinker.icu/vllm-project/vllm.git synced 2025-12-24 16:06:22 +08:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Roger Wang 2024-09-12 15:05:35 -07:00 committed by GitHub
parent c16369455f
commit b61bd98f90
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
tests/distributed/test_pipeline_parallel.py
View File

@ -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