mamba2

Model type

mamba2

Class

Mamba2CausalLMModel

Task

ssm2-text-generation

Source

models/mamba.py

Description

Mamba2/SSD causal language model with multi-head SSM.

Usage

mobius build --model <MODEL_ID> output_dir/
from mobius import build

model = build("<MODEL_ID>")