Unverified Commit 03b0364f authored by Ke Bao's avatar Ke Bao Committed by GitHub
Browse files

Update nextn ci test (#4071)

parent 2dd7d0c5
...@@ -58,7 +58,7 @@ class TestMLA(unittest.TestCase): ...@@ -58,7 +58,7 @@ class TestMLA(unittest.TestCase):
class TestDeepseekV3(unittest.TestCase): class TestDeepseekV3(unittest.TestCase):
@classmethod @classmethod
def setUpClass(cls): 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 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:
...@@ -93,7 +93,7 @@ class TestDeepseekV3(unittest.TestCase): ...@@ -93,7 +93,7 @@ class TestDeepseekV3(unittest.TestCase):
class TestDeepseekV3MTP(unittest.TestCase): class TestDeepseekV3MTP(unittest.TestCase):
@classmethod @classmethod
def setUpClass(cls): 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 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:
...@@ -106,9 +106,9 @@ class TestDeepseekV3MTP(unittest.TestCase): ...@@ -106,9 +106,9 @@ class TestDeepseekV3MTP(unittest.TestCase):
"--torch-compile-max-bs", "--torch-compile-max-bs",
"1", "1",
"--speculative-algorithm", "--speculative-algorithm",
"NEXTN", "EAGLE",
"--speculative-draft", "--speculative-draft",
"sgl-project/sglang-ci-dsv3-test-NextN", "lmsys/sglang-ci-dsv3-test-NextN",
"--speculative-num-steps", "--speculative-num-steps",
"2", "2",
"--speculative-eagle-topk", "--speculative-eagle-topk",
......
...@@ -17,7 +17,7 @@ from sglang.test.test_utils import ( ...@@ -17,7 +17,7 @@ from sglang.test.test_utils import (
class TestFlashinferMLA(unittest.TestCase): class TestFlashinferMLA(unittest.TestCase):
@classmethod @classmethod
def setUpClass(cls): 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 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:
...@@ -59,7 +59,7 @@ class TestFlashinferMLA(unittest.TestCase): ...@@ -59,7 +59,7 @@ class TestFlashinferMLA(unittest.TestCase):
class TestFlashinferMLANoRagged(unittest.TestCase): class TestFlashinferMLANoRagged(unittest.TestCase):
@classmethod @classmethod
def setUpClass(cls): 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 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:
......
...@@ -15,7 +15,7 @@ from sglang.test.test_utils import ( ...@@ -15,7 +15,7 @@ from sglang.test.test_utils import (
class TestDeepseekTP2(unittest.TestCase): class TestDeepseekTP2(unittest.TestCase):
@classmethod @classmethod
def setUpClass(cls): 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 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:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment