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

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