minke.yu/sglang
1
0
Fork 0
Code Issues Pull Requests Actions 12 Packages Projects Releases Wiki Activity

test: point DSV3 int8 MLA CI models to lmsys Hugging Face org (#21561)

Browse Source
This commit is contained in:
Lianmin Zheng
2026-03-27 13:04:01 -07:00
committed by GitHub
parent d864622a68
commit 9a91323c9f
1 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
test/registered/mla/test_mla_int8_deepseek_v3.py
+5 -5
View File
@@ -22,7 +22,7 @@ register_cuda_ci(est_time=341, suite="stage-b-test-1-gpu-large")
class TestMLADeepseekV3ChannelInt8(CustomTestCase): class TestMLADeepseekV3ChannelInt8(CustomTestCase):
@classmethod @classmethod
def setUpClass(cls): def setUpClass(cls):
cls.model = "sgl-project/sglang-ci-dsv3-channel-int8-test" cls.model = "lmsys/sglang-ci-dsv3-channel-int8-test"
cls.base_url = DEFAULT_URL_FOR_TEST cls.base_url = DEFAULT_URL_FOR_TEST
other_args = ["--trust-remote-code"] other_args = ["--trust-remote-code"]
if torch.cuda.is_available() and torch.version.cuda: if torch.cuda.is_available() and torch.version.cuda:
@@ -66,7 +66,7 @@ class TestMLADeepseekV3ChannelInt8(CustomTestCase):
class TestDeepseekV3MTPChannelInt8(CustomTestCase): class TestDeepseekV3MTPChannelInt8(CustomTestCase):
@classmethod @classmethod
def setUpClass(cls): def setUpClass(cls):
cls.model = "sgl-project/sglang-ci-dsv3-channel-int8-test" cls.model = "lmsys/sglang-ci-dsv3-channel-int8-test"
cls.base_url = DEFAULT_URL_FOR_TEST cls.base_url = DEFAULT_URL_FOR_TEST
other_args = ["--trust-remote-code"] other_args = ["--trust-remote-code"]
if torch.cuda.is_available() and torch.version.cuda: if torch.cuda.is_available() and torch.version.cuda:
@@ -80,7 +80,7 @@ class TestDeepseekV3MTPChannelInt8(CustomTestCase):
"--speculative-algorithm", "--speculative-algorithm",
"EAGLE", "EAGLE",
"--speculative-draft-model-path", "--speculative-draft-model-path",
"sgl-project/sglang-ci-dsv3-channel-int8-test-NextN", "lmsys/sglang-ci-dsv3-channel-int8-test-NextN",
"--speculative-num-steps", "--speculative-num-steps",
"2", "2",
"--speculative-eagle-topk", "--speculative-eagle-topk",
@@ -129,7 +129,7 @@ class TestDeepseekV3MTPChannelInt8(CustomTestCase):
class TestMLADeepseekV3BlockInt8(CustomTestCase): class TestMLADeepseekV3BlockInt8(CustomTestCase):
@classmethod @classmethod
def setUpClass(cls): def setUpClass(cls):
cls.model = "sgl-project/sglang-ci-dsv3-block-int8-test" cls.model = "lmsys/sglang-ci-dsv3-block-int8-test"
cls.base_url = DEFAULT_URL_FOR_TEST cls.base_url = DEFAULT_URL_FOR_TEST
other_args = ["--trust-remote-code"] other_args = ["--trust-remote-code"]
if torch.cuda.is_available() and torch.version.cuda: if torch.cuda.is_available() and torch.version.cuda:
@@ -172,7 +172,7 @@ class TestMLADeepseekV3BlockInt8(CustomTestCase):
class TestDeepseekV3MTPBlockInt8(CustomTestCase): class TestDeepseekV3MTPBlockInt8(CustomTestCase):
@classmethod @classmethod
def setUpClass(cls): def setUpClass(cls):
cls.model = "sgl-project/sglang-ci-dsv3-block-int8-test" cls.model = "lmsys/sglang-ci-dsv3-block-int8-test"
cls.base_url = DEFAULT_URL_FOR_TEST cls.base_url = DEFAULT_URL_FOR_TEST
other_args = ["--trust-remote-code"] other_args = ["--trust-remote-code"]
if torch.cuda.is_available() and torch.version.cuda: if torch.cuda.is_available() and torch.version.cuda: