diff --git a/vllm/v1/core/kv_cache_utils.py b/vllm/v1/core/kv_cache_utils.py index fab52a355d..04c5f5883a 100644 --- a/vllm/v1/core/kv_cache_utils.py +++ b/vllm/v1/core/kv_cache_utils.py @@ -645,12 +645,9 @@ def resolve_kv_cache_block_sizes( if not (cache_config.enable_prefix_caching or connector_enabled): return scheduler_block_size, scheduler_block_size - # Mamba groups with block_size != cache_config.block_size - # (mamba_cache_mode != "align") break divisibility; back off to the - # scheduler block size. if any( isinstance(g.kv_cache_spec, MambaSpec) - and g.kv_cache_spec.block_size != cache_config.block_size + and g.kv_cache_spec.mamba_cache_mode != "align" for g in groups ): return scheduler_block_size, scheduler_block_size