61 lines
2.1 KiB
Markdown
61 lines
2.1 KiB
Markdown
---
|
|
license: apache-2.0
|
|
base_model: mistralai/Mistral-7B-v0.3
|
|
tags:
|
|
- quantization
|
|
- ternary
|
|
- balanced-ternary
|
|
- tritllm
|
|
library_name: transformers
|
|
---
|
|
|
|
# Mistral-7B-v0.3-trit-uniform-d2
|
|
|
|
Balanced ternary quantization of [`mistralai/Mistral-7B-v0.3`](https://huggingface.co/mistralai/Mistral-7B-v0.3) at depth **d=2** (9 levels per weight, **3.47 bits per weight**).
|
|
|
|
Produced with the codec from **"Balanced Ternary Post-Training Quantization for Large Language Models"** (Stentzel, 2026). See [Entrit/tritllm-codec](https://huggingface.co/Entrit/tritllm-codec) for the codec source.
|
|
|
|
## Quick load
|
|
|
|
```python
|
|
from transformers import AutoModelForCausalLM, AutoTokenizer
|
|
|
|
model = AutoModelForCausalLM.from_pretrained("Entrit/Mistral-7B-v0.3-trit-uniform-d2")
|
|
tokenizer = AutoTokenizer.from_pretrained("Entrit/Mistral-7B-v0.3-trit-uniform-d2")
|
|
```
|
|
|
|
The weights are dequantized to FP16 for stock-`transformers` compatibility. The on-disk size is therefore the same as the FP16 source. The 3.47-bpw figure refers to the *information content* of the quantized matrices and is what matters for inference on hardware that consumes the packed trit format directly (see [Entrit/tritllm-kernel](https://huggingface.co/Entrit/tritllm-kernel)).
|
|
|
|
## Quantization details
|
|
|
|
| Field | Value |
|
|
|---|---|
|
|
| Source model | [`mistralai/Mistral-7B-v0.3`](https://huggingface.co/mistralai/Mistral-7B-v0.3) |
|
|
| Depth | d=2 (9 levels) |
|
|
| Bits per weight | 3.47 |
|
|
| Group size | 16 |
|
|
| Scale codebook | 27-entry log-spaced (scale_depth=3) |
|
|
| Method | Uniform PTQ |
|
|
| Quantized layers | all 2D linear matrices |
|
|
| Kept FP16 | `lm_head`, token embeddings, all `*_norm` layers |
|
|
| Codec | tritllm v2 |
|
|
|
|
## Citation
|
|
|
|
```
|
|
@article{stentzel2026ternaryptq,
|
|
title = {Balanced Ternary Post-Training Quantization for Large Language Models},
|
|
author = {Stentzel, Eric},
|
|
year = 2026,
|
|
note = {Entrit Systems}
|
|
}
|
|
```
|
|
|
|
## Reproducibility
|
|
|
|
```bash
|
|
git clone https://huggingface.co/Entrit/tritllm-codec
|
|
cd tritllm-codec
|
|
python quantize_model_v2.py --model mistralai/Mistral-7B-v0.3 --configs uniform-d2 --out ./out
|
|
```
|