WillHeld commited on
Commit
5052e68
·
verified ·
1 Parent(s): 97841c5

Document Snowball long-context checkpoint

Browse files
Files changed (1) hide show
  1. README.md +30 -0
README.md ADDED
@@ -0,0 +1,30 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: openmdw-1.1
3
+ language:
4
+ - en
5
+ pipeline_tag: text-generation
6
+ library_name: vllm
7
+ tags:
8
+ - moe
9
+ - marin
10
+ - grug
11
+ - base-model
12
+ ---
13
+ # Snowball 67B-A2B, 262K context, QK 1.75, skew 4×
14
+
15
+ BF16 export of step 157,000 from the [Snowball long-context comparison](https://github.com/marin-community/marin/issues/8977).
16
+ The run extended context to 262,144 tokens from step 156,000 through step 157,000 with `qk_mult=1.75`.
17
+ Long-context documents were upsampled 4× during extension.
18
+
19
+ The model has 67B total parameters, approximately 2B active parameters per token, 26 layers, 256 experts with four selected per token, and a 128,256-token vocabulary.
20
+ This is a base model. The tokenizer's bundled chat template does not indicate instruction tuning; use text completions.
21
+
22
+ Serving requires the [Marin vLLM fork](https://github.com/marin-community/vllm), which registers `GrugMoeForCausalLM` / `grug_moe`.
23
+ The existing serving configuration uses eight H100 GPUs, tensor parallelism 1, data parallelism 8, and expert parallelism because the model has five KV heads.
24
+ See the [serving and evaluation record](https://github.com/marin-community/marin/issues/8702) for the QK 1.57 and 1.75 baseline exports.
25
+ This upload validates tensor names, BF16 dtype, config, sizes, and SHA-256 checksums. No new generation or numerical parity test has been run for this upload.
26
+
27
+ The [export manifest](export-manifest.json) records each file’s size and SHA-256 checksum, the exporter revision, and the verified Hub weight revision.
28
+ The exporter applies pending QB router biases before conversion. Optimizer state is excluded.
29
+
30
+ Model materials are licensed under [OpenMDW 1.1](LICENSE).