Add default enable_memory_saver to HybridLinearKVPool (#13371)
This commit is contained in:
@@ -952,7 +952,7 @@ class HybridLinearKVPool(KVCache):
|
|||||||
enable_kvcache_transpose: bool,
|
enable_kvcache_transpose: bool,
|
||||||
device: str,
|
device: str,
|
||||||
mamba_pool: MambaPool,
|
mamba_pool: MambaPool,
|
||||||
enable_memory_saver: bool,
|
enable_memory_saver: bool = False,
|
||||||
# TODO: refactor mla related args
|
# TODO: refactor mla related args
|
||||||
use_mla: bool = False,
|
use_mla: bool = False,
|
||||||
kv_lora_rank: int = None,
|
kv_lora_rank: int = None,
|
||||||
|
|||||||
Reference in New Issue
Block a user