---
license: apache-2.0
language:
- en
tags:
- mamba
- mlx
- cartesia
---

# Model Card for mamba2-1.3b-4bit-mlx

This is an [MLX](https://ml-explore.github.io/mlx)-compatible version of the [mamba2-1.3b](https://huggingface.co/state-spaces/mamba2-1.3b) model, quantized to 4 bits. It uses the [EleutherAI/gpt-neox-20b](https://huggingface.co/EleutherAI/gpt-neox-20b) tokenizer.
For more details, see our [blog post](https://cartesia.ai/blog/on-device).

## Usage
### Installation
This model requires the `cartesia-metal` and `cartesia-mlx` packages.

Installation requires Xcode, which can be downloaded from https://developer.apple.com/xcode/. Accept the license agreement with:
```shell 
sudo xcodebuild -license
```

Install the required dependencies: the exact version of `nanobind`, followed by `cartesia-metal`, and finally `cartesia-mlx`, with the following commands:
```shell 
pip install nanobind@git+https://github.com/wjakob/nanobind.git@2f04eac452a6d9142dedb957701bdb20125561e4
pip install git+https://github.com/cartesia-ai/edge.git#subdirectory=cartesia-metal
pip install cartesia-mlx
```

Note: This package has been tested on macOS Sonoma 14.1 with the M3 chip.

### Generation example
```python 
import mlx.core as mx
import cartesia_mlx as cmx

model = cmx.from_pretrained("cartesia-ai/mamba2-1.3b-4bit-mlx")
model.set_dtype(mx.float32)   

prompt = "Rene Descartes was"

print(prompt, end="", flush=True)
for text in model.generate(
    prompt,
    max_tokens=500,
    eval_every_n=5,
    verbose=True,
    top_p=0.99,
    temperature=0.85,
):
    print(text, end="", flush=True)
```

## About Cartesia
At [Cartesia](https://cartesia.ai/), we're building real-time multimodal intelligence for every device.