ci: migrate VLM tests to test/registered/vlm/ (#16415)
This commit is contained in:
+4
@@ -1,3 +1,7 @@
|
|||||||
|
from sglang.test.ci.ci_register import register_cuda_ci
|
||||||
|
|
||||||
|
register_cuda_ci(est_time=150, suite="stage-b-test-small-1-gpu")
|
||||||
|
|
||||||
"""
|
"""
|
||||||
Usage:
|
Usage:
|
||||||
python3 -m unittest test_vision_chunked_prefill.TestVisionChunkedPrefill.test_chunked_prefill
|
python3 -m unittest test_vision_chunked_prefill.TestVisionChunkedPrefill.test_chunked_prefill
|
||||||
+7
-2
@@ -1,3 +1,7 @@
|
|||||||
|
from sglang.test.ci.ci_register import register_cuda_ci
|
||||||
|
|
||||||
|
register_cuda_ci(est_time=890, suite="stage-b-test-small-1-gpu")
|
||||||
|
|
||||||
"""
|
"""
|
||||||
Usage:
|
Usage:
|
||||||
python3 -m unittest test_vision_openai_server.TestOpenAIVisionServer.test_mixed_batch
|
python3 -m unittest test_vision_openai_server.TestOpenAIVisionServer.test_mixed_batch
|
||||||
@@ -7,8 +11,9 @@ python3 -m unittest test_vision_openai_server.TestOpenAIVisionServer.test_multi_
|
|||||||
import unittest
|
import unittest
|
||||||
|
|
||||||
import openai
|
import openai
|
||||||
from test_vision_openai_server_common import *
|
|
||||||
from test_vision_openai_server_common import ( # DEFAULT_TIMEOUT_FOR_SERVER_LAUNCH,; DEFAULT_URL_FOR_TEST,; IMAGE_MAN_IRONING_URL,; popen_launch_server,
|
from sglang.test.vlm_utils import *
|
||||||
|
from sglang.test.vlm_utils import (
|
||||||
AudioOpenAITestMixin,
|
AudioOpenAITestMixin,
|
||||||
CustomTestCase,
|
CustomTestCase,
|
||||||
ImageOpenAITestMixin,
|
ImageOpenAITestMixin,
|
||||||
@@ -1,3 +1,7 @@
|
|||||||
|
from sglang.test.ci.ci_register import register_cuda_ci
|
||||||
|
|
||||||
|
register_cuda_ci(est_time=430, suite="stage-b-test-small-1-gpu")
|
||||||
|
|
||||||
import json
|
import json
|
||||||
import unittest
|
import unittest
|
||||||
from io import BytesIO
|
from io import BytesIO
|
||||||
@@ -76,9 +76,6 @@ suites = {
|
|||||||
TestFile("test_torchao.py", 103),
|
TestFile("test_torchao.py", 103),
|
||||||
TestFile("test_utils_update_weights.py", 29),
|
TestFile("test_utils_update_weights.py", 29),
|
||||||
TestFile("test_video_utils.py", 5),
|
TestFile("test_video_utils.py", 5),
|
||||||
TestFile("test_vision_chunked_prefill.py", 150),
|
|
||||||
TestFile("test_vision_openai_server_a.py", 778),
|
|
||||||
TestFile("test_vlm_input_format.py", 166),
|
|
||||||
TestFile("test_modelopt_export.py", 9),
|
TestFile("test_modelopt_export.py", 9),
|
||||||
],
|
],
|
||||||
"per-commit-2-gpu": [
|
"per-commit-2-gpu": [
|
||||||
@@ -157,7 +154,6 @@ suites = {
|
|||||||
TestFile("test_eval_accuracy_large.py"),
|
TestFile("test_eval_accuracy_large.py"),
|
||||||
TestFile("test_gpt_oss_common.py"),
|
TestFile("test_gpt_oss_common.py"),
|
||||||
TestFile("test_moe_eval_accuracy_large.py"),
|
TestFile("test_moe_eval_accuracy_large.py"),
|
||||||
TestFile("test_vision_openai_server_common.py"),
|
|
||||||
TestFile("test_profile_v2.py"),
|
TestFile("test_profile_v2.py"),
|
||||||
TestFile("models/test_ministral3_models.py"),
|
TestFile("models/test_ministral3_models.py"),
|
||||||
TestFile("test_mistral_large3_basic.py"),
|
TestFile("test_mistral_large3_basic.py"),
|
||||||
|
|||||||
Reference in New Issue
Block a user