Add Qwen3-30B-A3B OSCAR KV rotations
Browse files
Qwen3-30B-A3B/README.md
ADDED
|
@@ -0,0 +1,22 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# Qwen3-30B-A3B OSCAR KV rotations
|
| 2 |
+
|
| 3 |
+
This model needs **per-head** rotations. One rotation shared across its four KV
|
| 4 |
+
heads collapses long generations (issue #16): the heads want mutually
|
| 5 |
+
near-orthogonal rotations (`mean |diag(R0^T R1)| ~= 0.07`).
|
| 6 |
+
|
| 7 |
+
| file | format | contents |
|
| 8 |
+
|---|---|---|
|
| 9 |
+
| `k_rotation_perhead.pt`, `v_rotation_perhead.pt` | `format_version: 2` | `rotation: [4, 128, 128]` per layer, 48 layers -- **use these** |
|
| 10 |
+
| `k_rotation_qqt_r_h_pbr.pt`, `v_rotation_sst_r_h_pbr.pt` | `format_version: 1` | one shared `[128, 128]` per layer -- for comparison only |
|
| 11 |
+
|
| 12 |
+
3 seeds, TP=2, 32K generation:
|
| 13 |
+
|
| 14 |
+
| Benchmark | BF16 | INT2 shared | INT2 per-head |
|
| 15 |
+
|---|:---:|:---:|:---:|
|
| 16 |
+
| GPQA | 60.8 +- 3.2 | 43.9 +- 2.8 | **58.6 +- 0.9** |
|
| 17 |
+
| HumanEval | 90.9 +- 2.1 | 25.8 +- 2.5 | **89.0 +- 1.2** |
|
| 18 |
+
| AIME 2025 | 70.0 +- 6.7 | 18.9 +- 10.7 | **65.6 +- 1.9** |
|
| 19 |
+
| MATH500 | 93.6 +- 0.9 | 60.8 +- 2.2 | **92.1 +- 0.4** |
|
| 20 |
+
|
| 21 |
+
Serving per-head needs a runtime that reads `format_version: 2`; the pool logs
|
| 22 |
+
`[per-head: 4 kv heads]` at startup when it is active.
|
Qwen3-30B-A3B/k_rotation_perhead.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:cf819e3d5d437e44e0245c2da065482d27d642d2cd2a7672428f91e375d4898a
|
| 3 |
+
size 12597631
|
Qwen3-30B-A3B/k_rotation_qqt_r_h_pbr.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:ec433986663dffad5f6646ab686f0b74f5dbd6c3555eb32d3bd8fd432d584603
|
| 3 |
+
size 3198717
|
Qwen3-30B-A3B/v_rotation_perhead.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:1d72de2cfc02bfffd5562f88f38cda75dd83e680bc4e6ab40995183d5e89371f
|
| 3 |
+
size 12597631
|
Qwen3-30B-A3B/v_rotation_sst_r_h_pbr.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:adaf48e0b85952e53baeeaf9d54f1e37f11101ba773f401ebc942f19700e7617
|
| 3 |
+
size 3198717
|