Upload Senryu-2.1-1B (EMA weights)
Browse files- README.md +5 -7
- tiny_gdn/web_fetch.py +1 -1
README.md
CHANGED
|
@@ -74,7 +74,7 @@ datasets:
|
|
| 74 |
[](https://www.apache.org/licenses/LICENSE-2.0)
|
| 75 |
[](https://huggingface.co/spaces/KerzgrAI/senryu-2.1-1b-demo)
|
| 76 |
|
| 77 |
-
*By [KerzgrAI](https://huggingface.co/KerzgrAI): [Senryu-2-1B](https://huggingface.co/KerzgrAI/Senryu-2-1B) trained on 2.06B more tokens, ending on a mix that is half new
|
| 78 |
|
| 79 |
</div>
|
| 80 |
|
|
@@ -84,7 +84,7 @@ datasets:
|
|
| 84 |
|
| 85 |
**Senryu-2.1-1B** is the second model on the Senryu-2 architecture. It continues [Senryu-2-1B](https://huggingface.co/KerzgrAI/Senryu-2-1B) with the same architecture, tokenizer and chat template, so it is a drop-in replacement.
|
| 86 |
|
| 87 |
-
- **Improved reasoning.**
|
| 88 |
- **Much better system prompt following.** 119,137 of those conversations were answered under their own system prompts (personas, output formats, rules and constraints from Multifaceted-Collection, SystemChat 2.0 and SystemChat 1.1), and the earlier steps added system-prompt datasets such as SystemCheck.
|
| 89 |
- **A default system prompt and default tools.** 36,929 conversations trained it under one Senryu system prompt with three tools, `web-search`, `fetch-page` and `calculator`, and rules to call them only when needed. `inference.py` and the [demo](https://huggingface.co/spaces/KerzgrAI/senryu-2.1-1b-demo) start every chat with both.
|
| 90 |
|
|
@@ -256,12 +256,12 @@ Interactive commands: `/think` `/no_think` `/system …` `/reset` `/exit`. Ctrl+
|
|
| 256 |
Continued from Senryu-2-1B's final SFT checkpoint (EMA weights) for 3,920 steps of 16 sequences × 32,768 tokens: 2.06B tokens, 1.41B of them assistant targets, in about 50 hours on one RTX PRO 6000 Blackwell GPU. Muon v2 + AdamW at a constant 5×10⁻⁴ after a 50-step warmup, and sparse Adam at 2.5×10⁻³ for the n-gram tables.
|
| 257 |
|
| 258 |
1. **Steps 1–2,230:** Senryu-2-1B's SFT mixture, refreshed with more system-prompt, chat and agentic data.
|
| 259 |
-
2. **Steps 2,230–3,920:** half new
|
| 260 |
|
| 261 |
| Part | Conversations | Tokens | What it is |
|
| 262 |
|------|--------------:|-------:|------------|
|
| 263 |
-
| System prompts | 119,137 | 225.0M |
|
| 264 |
-
| Default prompt and tools | 36,929 | 188.3M |
|
| 265 |
| Replay | 133,910 | 413.3M | The refreshed SFT mixture from steps 1–2,230, filtered of other models' self-descriptions |
|
| 266 |
|
| 267 |
Only assistant turns are trained. The released EMA weights reach a loss of 1.433 on the final mix's held-out split, against 1.498 for the raw weights. `sft_config.json` holds the run's settings.
|
|
@@ -285,8 +285,6 @@ Only assistant turns are trained. The released EMA weights reach a loss of 1.433
|
|
| 285 |
## Limitations
|
| 286 |
|
| 287 |
- **Scale.** 1.09B core parameters and no pretraining stage: expect thin world knowledge, factual errors and invented details.
|
| 288 |
-
- **Tool calls.** The model calls tools much less often than it should. It used the calculator reliably in our tests, but asked for this week's news, told outright to search the web, or given a link to read, it answered from memory and invented the details in 9 of 11 tries. Only 1.5% of its default-prompt training conversations search the web and 0.3% fetch a page. Paste a page's text in when it matters.
|
| 289 |
-
- **Search.** When it does search, it can misread or over-trust the results, and it sometimes repeats a search or lists results in a loop. Check anything that matters.
|
| 290 |
- **Language.** Training data is mostly English.
|
| 291 |
- **Hardware.** A CUDA GPU is required.
|
| 292 |
|
|
|
|
| 74 |
[](https://www.apache.org/licenses/LICENSE-2.0)
|
| 75 |
[](https://huggingface.co/spaces/KerzgrAI/senryu-2.1-1b-demo)
|
| 76 |
|
| 77 |
+
*By [KerzgrAI](https://huggingface.co/KerzgrAI): [Senryu-2-1B](https://huggingface.co/KerzgrAI/Senryu-2-1B) trained on 2.06B more tokens, ending on a mix that is half new reasoning, system-prompt and tool data*
|
| 78 |
|
| 79 |
</div>
|
| 80 |
|
|
|
|
| 84 |
|
| 85 |
**Senryu-2.1-1B** is the second model on the Senryu-2 architecture. It continues [Senryu-2-1B](https://huggingface.co/KerzgrAI/Senryu-2-1B) with the same architecture, tokenizer and chat template, so it is a drop-in replacement.
|
| 86 |
|
| 87 |
+
- **Improved reasoning.** Half of its last 827M training tokens are new conversations with a full reasoning trace in each turn's `<think>` block.
|
| 88 |
- **Much better system prompt following.** 119,137 of those conversations were answered under their own system prompts (personas, output formats, rules and constraints from Multifaceted-Collection, SystemChat 2.0 and SystemChat 1.1), and the earlier steps added system-prompt datasets such as SystemCheck.
|
| 89 |
- **A default system prompt and default tools.** 36,929 conversations trained it under one Senryu system prompt with three tools, `web-search`, `fetch-page` and `calculator`, and rules to call them only when needed. `inference.py` and the [demo](https://huggingface.co/spaces/KerzgrAI/senryu-2.1-1b-demo) start every chat with both.
|
| 90 |
|
|
|
|
| 256 |
Continued from Senryu-2-1B's final SFT checkpoint (EMA weights) for 3,920 steps of 16 sequences × 32,768 tokens: 2.06B tokens, 1.41B of them assistant targets, in about 50 hours on one RTX PRO 6000 Blackwell GPU. Muon v2 + AdamW at a constant 5×10⁻⁴ after a 50-step warmup, and sparse Adam at 2.5×10⁻³ for the n-gram tables.
|
| 257 |
|
| 258 |
1. **Steps 1–2,230:** Senryu-2-1B's SFT mixture, refreshed with more system-prompt, chat and agentic data.
|
| 259 |
+
2. **Steps 2,230–3,920:** half new data, half replay of that mixture. The final mix, 289,976 conversations and 826.6M tokens, was trained once through:
|
| 260 |
|
| 261 |
| Part | Conversations | Tokens | What it is |
|
| 262 |
|------|--------------:|-------:|------------|
|
| 263 |
+
| System prompts | 119,137 | 225.0M | Multifaceted-Collection, SystemChat 2.0 and SystemChat 1.1 openings, answered with full reasoning under their own system prompts |
|
| 264 |
+
| Default prompt and tools | 36,929 | 188.3M | smol-magpie-ultra openings, answered with full reasoning as Senryu-2.1 under the default system prompt and the three default tools; a quarter go on to a follow-up user turn |
|
| 265 |
| Replay | 133,910 | 413.3M | The refreshed SFT mixture from steps 1–2,230, filtered of other models' self-descriptions |
|
| 266 |
|
| 267 |
Only assistant turns are trained. The released EMA weights reach a loss of 1.433 on the final mix's held-out split, against 1.498 for the raw weights. `sft_config.json` holds the run's settings.
|
|
|
|
| 285 |
## Limitations
|
| 286 |
|
| 287 |
- **Scale.** 1.09B core parameters and no pretraining stage: expect thin world knowledge, factual errors and invented details.
|
|
|
|
|
|
|
| 288 |
- **Language.** Training data is mostly English.
|
| 289 |
- **Hardware.** A CUDA GPU is required.
|
| 290 |
|
tiny_gdn/web_fetch.py
CHANGED
|
@@ -1,7 +1,7 @@
|
|
| 1 |
"""Execute the `fetch-page` tool: read one web page and return its text as Markdown.
|
| 2 |
|
| 3 |
`fetch-page` sits next to `web-search` and `calculator` on the website and in
|
| 4 |
-
the tool-use
|
| 5 |
reduced to its main content by trafilatura, PDFs to their text, and plain
|
| 6 |
text types pass through. Each observation holds at most `PAGE_CHARS`
|
| 7 |
characters; a longer page ends with a note naming the `offset` that reads on.
|
|
|
|
| 1 |
"""Execute the `fetch-page` tool: read one web page and return its text as Markdown.
|
| 2 |
|
| 3 |
`fetch-page` sits next to `web-search` and `calculator` on the website and in
|
| 4 |
+
the tool-use training data, so a model sees the same observation in both. HTML is
|
| 5 |
reduced to its main content by trafilatura, PDFs to their text, and plain
|
| 6 |
text types pass through. Each observation holds at most `PAGE_CHARS`
|
| 7 |
characters; a longer page ends with a note naming the `offset` that reads on.
|