[CI] Remove duplicate code in test_mamba_ut (#16854)
Co-authored-by: luoyuan.luo <luoyuan.luo@antgroup.com>
This commit is contained in:
@@ -20,10 +20,7 @@ register_cuda_ci(est_time=9, suite="stage-b-test-small-1-gpu")
|
|||||||
class TestMamba(unittest.TestCase):
|
class TestMamba(unittest.TestCase):
|
||||||
@classmethod
|
@classmethod
|
||||||
def setUpClass(cls):
|
def setUpClass(cls):
|
||||||
server_args = ServerArgs(model_path="dummy")
|
pass
|
||||||
server_args.enable_dp_attention = False
|
|
||||||
server_args.page_size = 1
|
|
||||||
set_global_server_args_for_scheduler(server_args)
|
|
||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
def tearDownClass(cls):
|
def tearDownClass(cls):
|
||||||
|
|||||||
Reference in New Issue
Block a user