File size: 5,967 Bytes
23d844a
dd754d3
23d844a
 
 
 
 
 
 
 
 
 
dd754d3
23d844a
 
 
 
dd754d3
23d844a
 
 
 
dd754d3
 
 
825409e
9d4d8de
23d844a
9d4d8de
dd754d3
9d4d8de
23d844a
9d4d8de
dd754d3
9d4d8de
23d844a
dd754d3
 
 
 
 
9d4d8de
 
23d844a
dd754d3
80cde94
dd754d3
 
 
 
 
 
 
 
 
 
 
80cde94
dd754d3
352139c
dd754d3
 
 
 
23d844a
dd754d3
23d844a
dd754d3
23d844a
dd754d3
 
 
 
23d844a
dd754d3
23d844a
dd754d3
 
 
 
 
23d844a
dd754d3
23d844a
dd754d3
 
23d844a
 
 
 
dd754d3
23d844a
dd754d3
23d844a
dd754d3
 
 
 
 
 
 
 
 
 
 
23d844a
 
 
dd754d3
 
 
 
 
 
 
 
 
 
 
 
 
 
23d844a
 
 
 
 
dd754d3
23d844a
 
dd754d3
23d844a
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
---
language: en
license: apache-2.0
base_model: Zyphra/Zamba2-7B-instruct
tags:
  - zamba2
  - mamba
  - hybrid
  - compressed
  - hxq
  - helix-substrate
  - vector-quantization
  - helixcode
library_name: transformers
pipeline_tag: text-generation
model-index:
  - name: zamba2-7b-instruct-hxq
    results: []
---

# Zamba2-7B-Instruct-HXQ

> **2.0x smaller from BF16. 81-layer hybrid Mamba2+Transformer. Largest HXQ hybrid model.**
>
> Zamba2-7B-Instruct compressed from 14.7 GB (BF16) to 7.5 GB. 213 linear layers compressed, 573 exact tensors preserved. No calibration data. Just `pip install` and `from_pretrained()`.

## Install and Run

```bash
pip install "helix-substrate[hf]"
```

```python
import helix_substrate  # registers the HXQ quantizer with HuggingFace
from transformers import AutoModelForCausalLM, AutoTokenizer

model = AutoModelForCausalLM.from_pretrained("EchoLabs33/zamba2-7b-instruct-hxq")
tokenizer = AutoTokenizer.from_pretrained("EchoLabs33/zamba2-7b-instruct-hxq")

inputs = tokenizer("Explain the theory of relativity in simple terms:", return_tensors="pt")
outputs = model.generate(**inputs, max_new_tokens=128)
print(tokenizer.decode(outputs[0], skip_special_tokens=True))
```

That's it. `import helix_substrate` registers the quantizer. `from_pretrained()` handles the rest automatically.

## Benchmark

| | Dense (BF16) | HXQ |
|---|---|---|
| **Size** | 14.7 GB | **7.5 GB** |
| **Perplexity** (WikiText-2) | pending | pending |
| **Compression ratio** | β€” | **2.0x** |
| **Compressed modules** | β€” | 213 HelixLinear layers |
| **Architecture** | Zamba2 (81 layers, Mamba2 + shared Transformer) | unchanged |

## Good to Know

- **GPU recommended** β€” 7.5 GB requires 10+ GB VRAM. Use `device_map="auto"` for multi-GPU.
- **Fine-tunable via LoRA** β€” compressed weights remain frozen, but LoRA adapters attach to each `HelixLinear` layer via `HelixLinearSTE`. See `helix-substrate` for training infrastructure.
- **Requires `helix-substrate`** β€” the quantizer is not built into transformers. You need `pip install "helix-substrate[hf]"`.
- **Requires `transformers >= 4.45`** β€” for Zamba2 architecture support.
- **`mamba-ssm` recommended** β€” without it, falls back to a slower sequential code path.
- **PPL pending** β€” requires cloud GPU eval (model doesn't fit on 4 GB T2000).

## What is HelixCode?

HelixCode is a universal weight compression codec based on vector quantization:

- Each weight matrix is replaced by a **256-entry codebook** (float32) + **uint8 index matrix** + optional **sidecar corrections** for outlier values
- The compressed form *is* the executable β€” `HelixLinear` performs `codebook[indices] @ x` directly, no decompression step
- Works on any `nn.Linear` regardless of architecture (Transformer, Mamba, MLP, CNN)
- **No calibration data required** β€” unlike GPTQ/AWQ, codebooks are fit from the weights alone

## How It Works

1. `import helix_substrate` registers the `hxq` quantizer with HuggingFace
2. `from_pretrained()` reads `quantization_config.quant_method = "hxq"` from `config.json`
3. The quantizer replaces 213 `nn.Linear` modules with `HelixLinear` shells before weight loading
4. Safetensors populates the codebook, indices, and sidecar buffers directly
5. The model runs in compressed form β€” no decompression needed

## Architecture Details

Zamba2-7B-Instruct is a hybrid architecture with:
- **81 total layers** (Mamba2 + shared Transformer hybrid)
- **hidden_size=3584**, **attention_hidden_size=7168**, **32 attention heads**
- **mamba_d_state=64**, **mamba_d_conv=4**
- **vocab_size=32000**

213 linear layers compressed (162 Mamba projections, 38 attention/MLP, 26 LoRA adapters). Normalization layers, embeddings, conv1d, and Mamba-specific parameters (A_log, D, dt_bias) are stored at full precision.

## Compression Receipt

```
Compressed modules:  213
Exact tensors:       573  (norms, embeddings, conv1d, A_log, D, dt_bias, LoRA)
Skip tensors:        243  (from original model)
Total keys:          1425
Dense size:          14.7 GB (BF16)
Compressed size:     7.5 GB
Compression ratio:   2.0x
PPL delta:           pending (cloud GPU eval)
Gate 1:              PASS (structural validation + SHA256)
```

## Companion Models

Same codec, same `pip install`, multiple architectures:

| Model | Architecture | Ratio | PPL Delta |
|-------|-------------|-------|-----------|
| [qwen2.5-14b-instruct-helix](https://huggingface.co/EchoLabs33/qwen2.5-14b-instruct-helix) | Transformer | 3.4x | pending |
| [qwen2.5-7b-instruct-helix](https://huggingface.co/EchoLabs33/qwen2.5-7b-instruct-helix) | Transformer | 2.2x | +6.34% |
| [qwen2.5-3b-instruct-helix](https://huggingface.co/EchoLabs33/qwen2.5-3b-instruct-helix) | Transformer | 1.6x | +0.69% |
| [qwen2.5-coder-3b-helix](https://huggingface.co/EchoLabs33/qwen2.5-coder-3b-helix) | Transformer (code) | 1.6x | +1.92% |
| [qwen2.5-coder-1.5b-instruct-helix](https://huggingface.co/EchoLabs33/qwen2.5-coder-1.5b-instruct-helix) | Transformer (code) | 2.4x | +1.63% |
| [tinyllama-1.1b-helix](https://huggingface.co/EchoLabs33/tinyllama-1.1b-helix) | Transformer | 4.0x | +0.78% |
| [zamba2-2.7b-instruct-helix](https://huggingface.co/EchoLabs33/zamba2-2.7b-instruct-helix) | Hybrid (Mamba2+Transformer) | 1.8x | +6.59% |
| [zamba2-1.2b-helix](https://huggingface.co/EchoLabs33/zamba2-1.2b-helix) | Hybrid (Mamba2+Transformer) | 1.7x | +2.90% |
| [mamba2-1.3b-helix](https://huggingface.co/EchoLabs33/mamba2-1.3b-helix) | Pure SSM (Mamba2) | 2.1x | +8.0% |
| [mamba-130m-helix](https://huggingface.co/EchoLabs33/mamba-130m-helix) | Pure SSM | 3.8x | +18.4% |

## Citation

```bibtex
@software{helix_substrate_2026,
  title={Helix Substrate: Universal Weight Compression via HelixCode},
  author={EchoLabs},
  year={2026},
  url={https://github.com/echo313unfolding/helix-substrate}
}
```

## License

Apache 2.0 (inherited from [Zyphra/Zamba2-7B-instruct](https://huggingface.co/Zyphra/Zamba2-7B-instruct)).