diff --git a/python/sglang/test/test_deterministic_utils.py b/python/sglang/test/test_deterministic_utils.py index 47f6f82d9..59e9e4aa4 100644 --- a/python/sglang/test/test_deterministic_utils.py +++ b/python/sglang/test/test_deterministic_utils.py @@ -23,9 +23,13 @@ class TestDeterministicBase(CustomTestCase): def get_server_args(cls): return COMMON_SERVER_ARGS + @classmethod + def get_model(cls): + return DEFAULT_MODEL + @classmethod def setUpClass(cls): - cls.model = DEFAULT_MODEL + cls.model = cls.get_model() cls.base_url = DEFAULT_URL_FOR_TEST if "--attention-backend" not in cls.get_server_args(): raise unittest.SkipTest("Skip the base test class")