sfanm commited on
Commit
e418055
·
verified ·
1 Parent(s): e8d95b3

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +41 -0
README.md ADDED
@@ -0,0 +1,41 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: other
3
+ language: en
4
+ library_name: transformers
5
+ pipeline_tag: text-generation
6
+ tags:
7
+ - nanochat
8
+ - nemotron
9
+ - from-scratch
10
+ - perlmutter
11
+ - gpt2-tokenizer
12
+ ---
13
+ # d24-sft-v3-olmo3-10b-wholedoc
14
+
15
+ v3 SFT chat model — 50B ClimbMix base + 10B whole-doc OLMo-3 midtrain (2x the 5B corpus).
16
+
17
+ nanochat-style **depth-24** decoder — 24 layers × 1536 hidden × 12 heads, SwiGLU / RoPE / RMSNorm, tied embeddings, GPT-2 BPE vocab (50304), **0.757B params**, 2048-token context.
18
+
19
+ **Lineage.** v3 pretrain (**50B ClimbMix**) → OLMo-3 Dolmino **whole-doc** midtrain (**10.58B tok**, 2x the 5B-wholedoc, all 24 components at true OLMo-3 proportions, long docs sliced to 2048-seq by the loader) → SFT (nanochat mix: SmolTalk + MMLU-aux + GSM8K + spelling + identity).
20
+
21
+ **Metrics.** GSM8K (greedy, full 1319): **7.28%** · SFT val lm-loss 0.153. 2x the whole-doc data over the 5B-wholedoc SFT (6.60%) = **+0.68pt** — modest, diminishing returns; still below the math-dense v2 (9.86%). Confirms midtrain mix composition (math density) >> data quantity for GSM8K.
22
+
23
+ ## Use (chat)
24
+
25
+ This is a **chat model** (ChatML). The turn terminator it emits is the literal string **`<|im_end|>`** — which is **not** the `eos_token_id` (`50256` = `<|endoftext|>`) and is not even a single token. You **must** stop on the `<|im_end|>` string or generation will not stop:
26
+
27
+ ```python
28
+ from transformers import AutoModelForCausalLM, AutoTokenizer
29
+ mid = "sfanm/d24-sft-v3-olmo3-10b-wholedoc"
30
+ tok = AutoTokenizer.from_pretrained(mid)
31
+ model = AutoModelForCausalLM.from_pretrained(mid, torch_dtype="bfloat16", device_map="auto")
32
+
33
+ msgs = [{"role": "user", "content": "Natalia sold clips to 48 friends in April and half as many in May. How many total?"}]
34
+ ids = tok.apply_chat_template(msgs, add_generation_prompt=True, return_tensors="pt").to(model.device)
35
+ out = model.generate(ids, max_new_tokens=512, do_sample=False, stop_strings=["<|im_end|>"], tokenizer=tok)
36
+ print(tok.decode(out[0, ids.shape[1]:], skip_special_tokens=False))
37
+ ```
38
+
39
+ > Without `stop_strings=["<|im_end|>"]` the model rambles to `max_new_tokens`: the configured `eos_token_id` (50256) is the GPT-2 *document* EOS, which a chat turn does not end with. For vLLM, pass `stop=["<|im_end|>"]`.
40
+
41
+ *Research checkpoint from a from-scratch nanochat-d24 replication (pretrain → midtrain → SFT → RL) on NERSC Perlmutter. Trained on third-party corpora (ClimbMix, FineMath, OpenMath, MetaMath, OpenThoughts, OLMo-3 Dolmino, SmolTalk, …) — see those datasets' licenses; provided as-is for research.*