File size: 2,360 Bytes
5bdba90
0a8e207
5bdba90
0a8e207
5bdba90
 
 
 
 
0a8e207
5bdba90
0a8e207
5bdba90
0a8e207
5bdba90
0a8e207
 
5bdba90
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
0a8e207
 
 
 
 
 
 
 
 
5bdba90
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
---
license: mit
base_model: microsoft/phi-4
base_model_relation: quantized
library_name: transformers
tags:
- dashq
- quantized
- post-training-quantization
- int3
---
![DASH-Q](https://raw.githubusercontent.com/JaeminK/dashq/main/assets/dashq_banner.png)

# phi-4-DASHQ-INT3-g128

> **DASH-Q** — Diagonal-Aware Shrinkage for Robust PTQ.
> `INT3` · group size 128 · **7.9347 GB** (from 29.3190 GB — **3.7x smaller**)

## Usage

```python
from transformers import AutoModelForCausalLM, AutoTokenizer

model = AutoModelForCausalLM.from_pretrained(
    "jkim96/phi-4-DASHQ-INT3-g128", trust_remote_code=True, device_map="cuda", dtype="auto"
)
tokenizer = AutoTokenizer.from_pretrained("jkim96/phi-4-DASHQ-INT3-g128")

messages = [{"role": "user", "content": "Explain 2-bit quantization in one sentence."}]
text = tokenizer.apply_chat_template(messages, add_generation_prompt=True, tokenize=False)
inputs = tokenizer(text, return_tensors="pt").to(model.device)
print(tokenizer.decode(model.generate(**inputs, max_new_tokens=256)[0]))
```

`trust_remote_code=True` is required: the checkpoint ships its quantized-layer
implementation (`modeling_dashq.py`) and Triton kernels (`dashq_kernel.py`).
Without Triton, or on CPU, it falls back to dequantize-and-matmul in PyTorch.

### Requirements

| Package | Minimum | Verified with |
| --- | --- | --- |
| `torch` | 2.4 | 2.12.1+cu130 |
| `transformers` | 5.8 | 5.9.0 |
| `triton` | 3.0 (Linux; bundled with CUDA builds of PyTorch) | 3.7.1 |
| `huggingface_hub` | 1.5 (pulled in by transformers) | 1.15.0 |

## Quantization

| Field | Value |
| --- | --- |
| Base model | `microsoft/phi-4` |
| Precision | INT3, group size 128 |
| Scale / zero dtype | float16 |
| Calibration | wikitext2, 128 samples x 2048 |
| Size | 7.9347 GB  ·  original 29.3190 GB  ·  3.7x compression |

## Benchmarks

Full zero-shot / few-shot results for every DASH-Q checkpoint:
**[github.com/JaeminK/dashq#benchmarks](https://github.com/JaeminK/dashq#benchmarks)**

## Evaluation

| Metric | Value |
| --- | ---: |
| `wikitext2_ppl` | 6.7848 |
| `zero-shot accuracy avg` | 68.9681 |
| `arc_challenge` | 55.6314 |
| `arc_easy` | 74.8737 |
| `commonsense_qa` | 76.4947 |
| `hellaswag` | 80.1235 |
| `lambada_openai` | 74.2674 |
| `openbookqa` | 45.0000 |
| `piqa` | 81.3928 |
| `truthfulqa_mc2` | 57.7914 |
| `winogrande` | 75.1381 |