Fill in training results and required attribution
Browse files
README.md
CHANGED
|
@@ -18,19 +18,46 @@ This model was asked to be published under my account, not the creators. The com
|
|
| 18 |
|
| 19 |
A small from-scratch transformer matching the [babble / booper](https://github.com/kowo-co/babble) architecture, pretrained on a streamed slice of [openbmb/Ultra-FineWeb-L1](https://huggingface.co/datasets/openbmb/Ultra-FineWeb-L1) (filtered English web text, Apache-2.0).
|
| 20 |
|
| 21 |
-
This is **stage-1 English pretraining only**. It is not the Discord-tuned chatbot.
|
| 22 |
-
|
| 23 |
-
## What
|
| 24 |
-
|
| 25 |
-
|
| 26 |
-
-
|
| 27 |
-
|
| 28 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 29 |
|
| 30 |
## Files
|
| 31 |
|
| 32 |
- `latest.pt` — checkpoint (`model` state dict, `config`, optimizer, step/token counts)
|
| 33 |
-
- `tokenizer.json` — BPE merge list
|
| 34 |
- `loss.jsonl` — per-checkpoint train/val loss, throughput, and samples
|
| 35 |
|
| 36 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 18 |
|
| 19 |
A small from-scratch transformer matching the [babble / booper](https://github.com/kowo-co/babble) architecture, pretrained on a streamed slice of [openbmb/Ultra-FineWeb-L1](https://huggingface.co/datasets/openbmb/Ultra-FineWeb-L1) (filtered English web text, Apache-2.0).
|
| 20 |
|
| 21 |
+
This is **stage-1 English pretraining only**. It is not the Discord-tuned chatbot. The architecture, tokenizer scheme, and training script come from [`kowo-co/babble`](https://github.com/kowo-co/babble) (`pretrain_hf.py` + the default 34.1M config).
|
| 22 |
+
|
| 23 |
+
## What it is
|
| 24 |
+
|
| 25 |
+
| | |
|
| 26 |
+
|---|---|
|
| 27 |
+
| Parameters | 34,096,128 |
|
| 28 |
+
| Layers / width / heads | 8 / 512 / 8 |
|
| 29 |
+
| Context | 1024 tokens |
|
| 30 |
+
| Tokenizer | byte-level BPE, 16,384 tokens |
|
| 31 |
+
| Data | `openbmb/Ultra-FineWeb-L1` |
|
| 32 |
+
| Train split | `CC-MAIN-2025-51` |
|
| 33 |
+
| Val split | `CC-MAIN-2025-47` (disjoint crawl) |
|
| 34 |
+
| Tokens trained | 600,206,202 |
|
| 35 |
+
| Hardware | 1× NVIDIA H200 (Hugging Face Jobs) |
|
| 36 |
+
| Wall clock | ~51 minutes end-to-end (~46 min of training at ~217k tok/s) |
|
| 37 |
+
| Final train loss | 2.531 |
|
| 38 |
+
| Final val loss | 2.472 |
|
| 39 |
+
|
| 40 |
+
Val loss fell steadily from 4.23 (step 200) to 2.47 (step 3118). Loss is **nats per BPE token**, not nats per byte, so it is not comparable to babble's older byte-level numbers without a bits-per-character conversion.
|
| 41 |
|
| 42 |
## Files
|
| 43 |
|
| 44 |
- `latest.pt` — checkpoint (`model` state dict, `config`, optimizer, step/token counts)
|
| 45 |
+
- `tokenizer.json` — BPE merge list, loadable with `babble.subword.BPETokenizer.from_json`
|
| 46 |
- `loss.jsonl` — per-checkpoint train/val loss, throughput, and samples
|
| 47 |
|
| 48 |
+
## End-of-run samples
|
| 49 |
+
|
| 50 |
+
Prompts used by the training script (temperature 0.7, top-k 40):
|
| 51 |
+
|
| 52 |
+
- `the cat` → `the catastrophic system and its adaptability to manage the catastrophic system is the case with a significant surge`
|
| 53 |
+
- `In the beginning` → `In the beginning of the New Jersey Law and Law, “law enforcement of the Law and Law and Law in the law`
|
| 54 |
+
- `Scientists have discovered` → `Scientists have discovered that the current market is expected to take a long way to see how we look at the new market`
|
| 55 |
+
- `The weather today is` → `The weather today is about to create a more sustainable and sustainable future. It’s a way to make the most of your time and`
|
| 56 |
+
|
| 57 |
+
These are expected to be clumsy: 34M params and 600M tokens is a short English pretrain, not a finished assistant.
|
| 58 |
+
|
| 59 |
+
## Training
|
| 60 |
+
|
| 61 |
+
Run on Hugging Face Jobs with the repo's self-contained `pretrain_hf.py` (bf16 AMP, AdamW, cosine LR after warmup). Job: [ProCreations/6a893d3e7c5c7dd37923450f](https://huggingface.co/jobs/ProCreations/6a893d3e7c5c7dd37923450f).
|
| 62 |
+
|
| 63 |
+
Source: [kowo-co/babble](https://github.com/kowo-co/babble), including [PR #21](https://github.com/kowo-co/babble/pull/21).
|