diff --git a/test/registered/attention/test_mamba_unittest.py b/test/registered/attention/test_mamba_unittest.py index 937c16baa..0a0f45fb8 100644 --- a/test/registered/attention/test_mamba_unittest.py +++ b/test/registered/attention/test_mamba_unittest.py @@ -20,10 +20,7 @@ register_cuda_ci(est_time=9, suite="stage-b-test-small-1-gpu") class TestMamba(unittest.TestCase): @classmethod def setUpClass(cls): - server_args = ServerArgs(model_path="dummy") - server_args.enable_dp_attention = False - server_args.page_size = 1 - set_global_server_args_for_scheduler(server_args) + pass @classmethod def tearDownClass(cls):