[Feature] Re:Enable hybrid mem saver (#12962)

2025-11-18 21:29:15 +08:00
parent 63807079b9
commit 518467be64
2 changed files with 14 additions and 10 deletions
@@ -41,6 +41,7 @@ class TestMamba(unittest.TestCase):
            full_attention_layer_ids=full_attention_layer_ids,
            enable_kvcache_transpose=False,
            device=device,
+            enable_memory_saver=False,
            mamba_pool=None,
        )
        assert pool._transfer_full_attention_id(global_interval - 1) == 0
@@ -173,6 +174,7 @@ class TestMamba(unittest.TestCase):
            full_attention_layer_ids=full_attention_layer_ids,
            enable_kvcache_transpose=False,
            device=device,
+            enable_memory_saver=False,
            mamba_pool=req_to_token_pool.mamba_pool,
        )