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):
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",
......
......@@ -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:
......
......@@ -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:
......
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