Update nextn ci test (#4071)

This commit is contained in:
Ke Bao
2025-03-05 05:01:24 +08:00
committed by GitHub
parent 2dd7d0c533
commit 03b0364f76
3 changed files with 7 additions and 7 deletions

View File

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

View File

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

View File

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