What is quantized
This is a hybrid model (mixed linear_attention / full_attention layers). The recipe
follows the proven approach for this architecture: quantize the standard projections, keep the
sensitive state-space gates and non-LLM parts in BF16.
Quantized to INT4 (200 Linear modules):
- full-attention
q_proj / k_proj / v_proj / o_proj (8 layers)
- all MLP
gate_proj / up_proj / down_proj (32 layers)
- linear-attention
out_proj, in_proj_qkv, in_proj_z (24 layers)
Kept in BF16 (ignore, 159 Linear modules):
- the entire vision tower —
re:model\.visual\..* (110)
- linear-attention SSM gates
in_proj_a / in_proj_b (48) — sensitive to the recurrence
lm_head, and any mtp head
Usage (vLLM)
vllm serve Ar4ikov/Ornith-1.0-9B-AWQ-W4A16-ASYM --trust-remote-code
Expected GPU memory ≈ checkpoint size + KV cache + activations.
Recommended sampling
The model emits <think>...</think> reasoning blocks before the final answer.
temperature = 0.6, top_p = 0.95, top_k = 20
Reproducibility
The recipe.yaml written by llm-compressor is included in this repo. config.json carries the
quantization_config (compressed-tensors, format pack-quantized) describing the exact
scheme, group size and ignore list.
Notes
- Quantizing
lm_head and the vision tower hurts quality for little gain, so they stay BF16.
Together with the large embedding table and the BF16 SSM gates, this is why a 9B hybrid
compresses to ~8 GB rather than the ~4.5 GB of a pure-transformer 7B.
- Loading this checkpoint in plain
transformers transparently decompresses the weights
back to BF16, so VRAM there looks like the original model. The real INT4 memory saving is
realized by vLLM's native 4-bit kernels.