state-spaces · First MLX build 21 Sept 2024

mamba-130m-hf

MLX Text

A 129M parameter text generation model from state-spaces, with 2 MLX builds at bf16.

Hugging Face
Parameters
129M
From Hugging Face
Quantisation
Unstated
mamba-130m-hf-f32
Memory needed
3 GB
Estimate, unified memory
Download
517 MB
Weights, estimated
Downloads
29
This build, last 30 days
Task
Text generation
Text
MLX builds
2
On mlx-community
Devices
Mac
Apple silicon

Memory

Estimated from the size of the weights, plus a fifth for context and working memory and 2 GB for macOS and your apps. Long contexts need more.

Run it

Terminal · mlx-lm
$ pip install -U mlx-lm
$ mlx_lm.generate --model mlx-community/mamba-130m-hf-f32 --prompt "Hello"

MLX builds

2 on mlx-community
Click a column to sort. Smallest size and most downloads are highlighted.
mamba-130m-hf-bf16 bf16 258 MB 3 GB 24
mamba-130m-hf-f32 Unstated 517 MB 3 GB 29

Sizes are estimated from Hugging Face’s parameter counts and the quantisation. Downloads are for the last 30 days.