Text Generation
MLX
Safetensors
English
pretraining
from-scratch
small-language-model
post-training
silicon
Instructions to use wzebrowski/OpenSML-150M with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- MLX
How to use wzebrowski/OpenSML-150M with MLX:
# Make sure mlx-lm is installed # pip install --upgrade mlx-lm # if on a CUDA device, also pip install mlx[cuda] # Generate text with mlx-lm from mlx_lm import load, generate model, tokenizer = load("wzebrowski/OpenSML-150M") prompt = "Once upon a time in" text = generate(model, tokenizer, prompt=prompt, verbose=True) - Notebooks
- Google Colab
- Kaggle
- Local Apps Settings
- LM Studio
- MLX LM
How to use wzebrowski/OpenSML-150M with MLX LM:
Generate or start a chat session
# Install MLX LM uv tool install mlx-lm # Generate some text mlx_lm.generate --model "wzebrowski/OpenSML-150M" --prompt "Once upon a time"
- Atomic Chat
Add Phase A and Phase B pretraining loss charts
Browse files
README.md
CHANGED
|
@@ -34,7 +34,7 @@ This repository describes one model, not a family of released models.
|
|
| 34 |
|
| 35 |
**Selected checkpoint:** Stage B step 73,243 路 **Exposure:** 7,800,086,528 pretraining tokens 路 **Status:** documentation preview; weight release pending.
|
| 36 |
|
| 37 |
-
[Model specification](#model-specification) 路 [Training method](#training-method) 路 [Evaluation](#evaluation) 路 [Limitations](#limitations-and-intended-use)
|
| 38 |
|
| 39 |
## Availability and Inference
|
| 40 |
|
|
@@ -228,6 +228,47 @@ This is **not a single unchanged 15B schedule from initialization**, and the
|
|
| 228 |
The linked [pilot](configs/pilot.json) and [early continuation configuration](configs/training_recipe.json)
|
| 229 |
are historical recipes; the later transitions are summarized here.
|
| 230 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 231 |
## Interim Results
|
| 232 |
|
| 233 |
**Dated snapshot: September 18, 2026, through step 8,217. Not a live counter.**
|
|
|
|
| 34 |
|
| 35 |
**Selected checkpoint:** Stage B step 73,243 路 **Exposure:** 7,800,086,528 pretraining tokens 路 **Status:** documentation preview; weight release pending.
|
| 36 |
|
| 37 |
+
[Model specification](#model-specification) 路 [Training method](#training-method) 路 [Pretraining curves](#pretraining-loss-curves) 路 [Evaluation](#evaluation) 路 [Limitations](#limitations-and-intended-use)
|
| 38 |
|
| 39 |
## Availability and Inference
|
| 40 |
|
|
|
|
| 228 |
The linked [pilot](configs/pilot.json) and [early continuation configuration](configs/training_recipe.json)
|
| 229 |
are historical recipes; the later transitions are summarized here.
|
| 230 |
|
| 231 |
+
## Pretraining Loss Curves
|
| 232 |
+
|
| 233 |
+
These charts show recorded held-out next-token validation loss from the selected
|
| 234 |
+
pretraining lineage, without smoothing. **Blue is Phase A** (the initial mixture
|
| 235 |
+
and its continuations); **teal is Phase B** (the Stage B prose refinement).
|
| 236 |
+
Phase B starts at step **73,008**, after **7.775B tokens**. The star marks the
|
| 237 |
+
selected pretrained base at step **73,243**, after **7.800B tokens**. Later points
|
| 238 |
+
show continued pretraining beyond the selected checkpoint, not additional
|
| 239 |
+
training included in that checkpoint. Lower loss is better.
|
| 240 |
+
|
| 241 |
+
### Full History and Late-training Zoom
|
| 242 |
+
|
| 243 |
+

|
| 244 |
+
|
| 245 |
+
The overview and zoom use the **same original held-out set** throughout
|
| 246 |
+
(262,144 target positions). The zoom makes the shorter Phase B visible; its
|
| 247 |
+
vertical scale is magnified. Aggregate validation weights remain fixed at
|
| 248 |
+
55% web, 25% DCLM, 10% Wikipedia, and 10% Cosmopedia across both phases,
|
| 249 |
+
even though the Phase B training mixture changes. The selected checkpoint's
|
| 250 |
+
loss on this original set is **2.81855024**.
|
| 251 |
+
|
| 252 |
+
### Phase B Detail
|
| 253 |
+
|
| 254 |
+

|
| 255 |
+
|
| 256 |
+
This close-up uses the **expanded held-out set** (4,194,304 target positions),
|
| 257 |
+
introduced late in Phase A. Its absolute loss values are separate from the
|
| 258 |
+
original-set curves above and below. The selected checkpoint's expanded loss
|
| 259 |
+
is **2.76700753**. The tightly zoomed vertical scale shows small changes; the
|
| 260 |
+
later increase is less than 0.001 loss. Validation weights remain fixed to the
|
| 261 |
+
original reference mixture.
|
| 262 |
+
|
| 263 |
+
### Validation Loss by Data Source
|
| 264 |
+
|
| 265 |
+

|
| 266 |
+
|
| 267 |
+
Each source is evaluated on the original held-out set throughout. Panels use
|
| 268 |
+
their own vertical scales; they show each source's trajectory rather than a
|
| 269 |
+
common-scale ranking of source difficulty. The dashed line marks the Phase B
|
| 270 |
+
transition and the star marks the selected base.
|
| 271 |
+
|
| 272 |
## Interim Results
|
| 273 |
|
| 274 |
**Dated snapshot: September 18, 2026, through step 8,217. Not a live counter.**
|