[v1][mamba] Added mamba_type into MambaSpec (#21715)

Signed-off-by: asafg <asafg@ai21.com> Co-authored-by: asafg <asafg@ai21.com>
2025-07-28 11:15:55 +03:00
parent 139a7f07bd
commit a6c050286a
6 changed files with 52 additions and 4 deletions
--- a/vllm/model_executor/layers/mamba/mamba_mixer2.py
+++ b/vllm/model_executor/layers/mamba/mamba_mixer2.py
@@ -732,6 +732,10 @@ class MambaMixer2(MambaBase, CustomOp):
            conv_kernel=self.conv_kernel_size,
        )

+    @property
+    def mamba_type(self) -> str:
+        return "mamba2"
+

 def mamba_mixer2(
    hidden_states: torch.Tensor,