ProCreations commited on
Commit
83d395e
·
verified ·
1 Parent(s): bb8ec26

Fill in training results and required attribution

Browse files
Files changed (1) hide show
  1. README.md +37 -10
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. Architecture, tokenizer scheme, and training script come from [`kowo-co/babble`](https://github.com/kowo-co/babble) (`pretrain_hf.py` + `configs/pretrain/default.json`).
22
-
23
- ## What this is
24
-
25
- - ~34.1M parameters (8 layers, 512-wide, 8 heads, context 1024)
26
- - Byte-level BPE tokenizer (16,384 tokens) fit on the train split
27
- - ~600M training tokens from Ultra-FineWeb-L1 (`CC-MAIN-2025-51` train, `CC-MAIN-2025-47` held-out val)
28
- - Trained on Hugging Face Jobs with the repo's self-contained `pretrain_hf.py`
 
 
 
 
 
 
 
 
 
 
 
 
29
 
30
  ## Files
31
 
32
  - `latest.pt` — checkpoint (`model` state dict, `config`, optimizer, step/token counts)
33
- - `tokenizer.json` — BPE merge list (`babble.subword.BPETokenizer.from_json`)
34
  - `loss.jsonl` — per-checkpoint train/val loss, throughput, and samples
35
 
36
- Training metrics and samples will be filled in here after the job finishes.
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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).