[Bugfix][MRV2] Forward seq_lens_cpu_upper_bound for mamba hybrid models (#46759)
Signed-off-by: mgoin <mgoin64@gmail.com>
Michael Goin committed
Jun 25, 2026 at 20:03 UTC
3daea7ceb990bff87e925b2f4b77325af052282f
1 file changed
+1
vllm/v1/worker/gpu/model_states/mamba_hybrid.py
+1
@@ -137,6 +137,7 @@ class MambaHybridModelState(DefaultModelState):
137
block_tables=block_tables,
138
slot_mappings=slot_mappings,
139
kv_cache_config=kv_cache_config,
140
+ seq_lens_cpu_upper_bound=seq_lens_cpu_upper_bound,
141
dcp_local_seq_lens=input_batch.dcp_local_seq_lens,
142
model_specific_attn_metadata=mamba_attn_metadata,
143
for_cudagraph_capture=for_capture,