diff --git a/test/srt/test_mla.py b/test/srt/test_mla.py index abc2fb656..a019988ab 100644 --- a/test/srt/test_mla.py +++ b/test/srt/test_mla.py @@ -58,7 +58,7 @@ class TestMLA(unittest.TestCase): class TestDeepseekV3(unittest.TestCase): @classmethod def setUpClass(cls): - cls.model = "sgl-project/sglang-ci-dsv3-test" + cls.model = "lmsys/sglang-ci-dsv3-test" cls.base_url = DEFAULT_URL_FOR_TEST other_args = ["--trust-remote-code"] if torch.cuda.is_available() and torch.version.cuda: @@ -93,7 +93,7 @@ class TestDeepseekV3(unittest.TestCase): class TestDeepseekV3MTP(unittest.TestCase): @classmethod def setUpClass(cls): - cls.model = "sgl-project/sglang-ci-dsv3-test" + cls.model = "lmsys/sglang-ci-dsv3-test" cls.base_url = DEFAULT_URL_FOR_TEST other_args = ["--trust-remote-code"] if torch.cuda.is_available() and torch.version.cuda: @@ -106,9 +106,9 @@ class TestDeepseekV3MTP(unittest.TestCase): "--torch-compile-max-bs", "1", "--speculative-algorithm", - "NEXTN", + "EAGLE", "--speculative-draft", - "sgl-project/sglang-ci-dsv3-test-NextN", + "lmsys/sglang-ci-dsv3-test-NextN", "--speculative-num-steps", "2", "--speculative-eagle-topk", diff --git a/test/srt/test_mla_flashinfer.py b/test/srt/test_mla_flashinfer.py index fc29e958f..04586acc5 100644 --- a/test/srt/test_mla_flashinfer.py +++ b/test/srt/test_mla_flashinfer.py @@ -17,7 +17,7 @@ from sglang.test.test_utils import ( class TestFlashinferMLA(unittest.TestCase): @classmethod def setUpClass(cls): - cls.model = "sgl-project/sglang-ci-dsv3-test" + cls.model = "lmsys/sglang-ci-dsv3-test" cls.base_url = DEFAULT_URL_FOR_TEST other_args = ["--trust-remote-code"] if torch.cuda.is_available() and torch.version.cuda: @@ -59,7 +59,7 @@ class TestFlashinferMLA(unittest.TestCase): class TestFlashinferMLANoRagged(unittest.TestCase): @classmethod def setUpClass(cls): - cls.model = "sgl-project/sglang-ci-dsv3-test" + cls.model = "lmsys/sglang-ci-dsv3-test" cls.base_url = DEFAULT_URL_FOR_TEST other_args = ["--trust-remote-code"] if torch.cuda.is_available() and torch.version.cuda: diff --git a/test/srt/test_mla_tp.py b/test/srt/test_mla_tp.py index a5f2420cc..777be82e0 100644 --- a/test/srt/test_mla_tp.py +++ b/test/srt/test_mla_tp.py @@ -15,7 +15,7 @@ from sglang.test.test_utils import ( class TestDeepseekTP2(unittest.TestCase): @classmethod def setUpClass(cls): - cls.model = "sgl-project/sglang-ci-dsv3-test" + cls.model = "lmsys/sglang-ci-dsv3-test" cls.base_url = DEFAULT_URL_FOR_TEST other_args = ["--trust-remote-code"] if torch.cuda.is_available() and torch.version.cuda: