Raghav-Singhal commited on
Commit
aaabfb0
·
verified ·
1 Parent(s): 6143ca5

Note SPP-annotated docs on the pretraining line

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -24,7 +24,7 @@ Trained with Synthetic Persona Pretraining (SPP) from token zero, then post-trai
24
  Base counterpart: [`model-raising/spp-t0-3b-base`](https://huggingface.co/model-raising/spp-t0-3b-base).
25
 
26
  ## Model details
27
- - **Architecture:** Llama-3.2-3B-shaped dense decoder-only transformer (28 layers, hidden size 3072, 24 attention heads, 8 KV groups, RoPE, RMSNorm, SwiGLU, tied input/output embeddings), trained from scratch.
28
  - **Tokenizer:** SmolLM2 tokenizer with an added `<assistant>` marker token (vocabulary 49280).
29
  - **Pretraining:** ~500B tokens on a subset of the Olmo 3 Dolma 3 mixture, with SPP reflections inserted into the safety-annotated documents within it.
30
  - **Post-training:** persona-binding supervised fine-tuning (PBSFT-mix): 300k single-turn examples, 90% WildChat-1M instructions and 10% safety prompts (WildJailbreak, WildGuardMix); assistant responses follow the Model Raising Constitution with inline `[N.M]` citations; response-only loss, one epoch.
 
24
  Base counterpart: [`model-raising/spp-t0-3b-base`](https://huggingface.co/model-raising/spp-t0-3b-base).
25
 
26
  ## Model details
27
+ - **Architecture:** Llama-3.2-3B-shaped, trained from scratch.
28
  - **Tokenizer:** SmolLM2 tokenizer with an added `<assistant>` marker token (vocabulary 49280).
29
  - **Pretraining:** ~500B tokens on a subset of the Olmo 3 Dolma 3 mixture, with SPP reflections inserted into the safety-annotated documents within it.
30
  - **Post-training:** persona-binding supervised fine-tuning (PBSFT-mix): 300k single-turn examples, 90% WildChat-1M instructions and 10% safety prompts (WildJailbreak, WildGuardMix); assistant responses follow the Model Raising Constitution with inline `[N.M]` citations; response-only loss, one epoch.