Update nextn ci test (#4071)
This commit is contained in:
@@ -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",
|
||||
|
||||
Reference in New Issue
Block a user