@tuned-tensor/cli 0.13.1-beta.4 → 0.13.1-beta.6
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/README.md +22 -8
- package/dist/{chunk-7VIUHQBI.js → chunk-W7MOBPCO.js} +1544 -814
- package/dist/chunk-W7MOBPCO.js.map +1 -0
- package/dist/index.js +275 -33
- package/dist/index.js.map +1 -1
- package/dist/local-runtime.js +99 -6
- package/dist/local-runtime.js.map +1 -1
- package/docs/local-runtime/architecture.md +10 -6
- package/docs/local-runtime/foundation-long-runs.md +108 -0
- package/docs/local-runtime/spark.md +6 -0
- package/package.json +1 -1
- package/training/foundation/src/checkpoint.py +235 -0
- package/training/foundation/src/common.py +30 -2
- package/training/foundation/src/data.py +89 -3
- package/training/foundation/src/evaluate.py +10 -2
- package/training/foundation/src/pretrain.py +426 -27
- package/training/foundation/src/train_tokenizer.py +29 -5
- package/dist/chunk-7VIUHQBI.js.map +0 -1
package/README.md
CHANGED
|
@@ -141,7 +141,7 @@ guarantee that a model or dataset is compatible with a TT workflow.
|
|
|
141
141
|
|
|
142
142
|
For educational questions about how or why training works, the agent can read
|
|
143
143
|
the exact Python implementation shipped with the running TT build. The source
|
|
144
|
-
tool is limited to named foundation components (tokenizer, pretraining, model,
|
|
144
|
+
tool is limited to named foundation components (tokenizer, pretraining, checkpointing, model,
|
|
145
145
|
data, SFT, RL, evaluation, and common helpers) and adapter components (training,
|
|
146
146
|
data, model contract, and evaluation); it cannot read arbitrary paths. Answers
|
|
147
147
|
separate behavior directly visible in the code from inferred rationale and
|
|
@@ -206,16 +206,18 @@ Two local engines share that document version:
|
|
|
206
206
|
`inference` evaluators. Requires a foundation `tunedtensor.json`
|
|
207
207
|
(`engine: "foundation"`, at least two examples, no `base_model`).
|
|
208
208
|
|
|
209
|
-
The foundation engine is a
|
|
209
|
+
The foundation engine is a readable, single-GPU local trainer inspired by Andrej
|
|
210
210
|
Karpathy's [nanochat](https://github.com/karpathy/nanochat), not a port of its
|
|
211
|
-
|
|
212
|
-
|
|
213
|
-
|
|
211
|
+
distributed training stack. With `foundation.corpus_path`, it streams `.txt`
|
|
212
|
+
or JSONL (`{"text":"..."}`) shards into a deterministic on-disk token cache;
|
|
213
|
+
without that field it retains the bounded prompt/example corpus for smoke runs.
|
|
214
|
+
`foundation.validation_path` supplies a distinct held-out BPB corpus. Foundation
|
|
214
215
|
plans reject `compare` before creating run artifacts; comparison remains an
|
|
215
216
|
adapter-engine capability. Optional RL samples one seeded on-policy completion
|
|
216
217
|
per step with a sparse last-number exact reward and therefore requires a numeric
|
|
217
218
|
expected output for every example. Treat those metrics as an end-to-end
|
|
218
|
-
execution and overfit check—not held-out capability or multi-GPU evidence
|
|
219
|
+
execution and overfit check—not held-out capability or multi-GPU evidence when
|
|
220
|
+
no validation corpus is configured.
|
|
219
221
|
|
|
220
222
|
```bash
|
|
221
223
|
tt pipeline init --file pipeline.json
|
|
@@ -227,6 +229,7 @@ tt --json pipeline run --file pipeline.json \
|
|
|
227
229
|
--spec tunedtensor.json --config local-runner.json
|
|
228
230
|
tt init --engine foundation --name "Tiny GPT"
|
|
229
231
|
tt --json pipeline run --spec tunedtensor.json --dry-run
|
|
232
|
+
tt pipeline run --spec tunedtensor.json --resume /absolute/path/to/foundation-run
|
|
230
233
|
```
|
|
231
234
|
|
|
232
235
|
`--only` and `--skip` preserve dependency safety: a selected step cannot refer
|
|
@@ -234,20 +237,31 @@ to an omitted predecessor. Cloud-targeted steps fail closed unless you pass
|
|
|
234
237
|
`--dry-run`. Foundation documents are local-only. The TT agent may describe,
|
|
235
238
|
validate, and prepare a plan, but has no direct pipeline-execute tool.
|
|
236
239
|
|
|
240
|
+
Long pretraining uses BF16, warmup plus cosine decay, gradient clipping,
|
|
241
|
+
periodic metrics, and atomic rolling checkpoints. `--resume <run-directory>`
|
|
242
|
+
starts or resumes one stable run identity; completed stages are verified and
|
|
243
|
+
skipped, while an interrupted pretrain step restores model, optimizer,
|
|
244
|
+
scheduler, RNG, counters, and token cursor. See
|
|
245
|
+
[Long foundation runs](docs/local-runtime/foundation-long-runs.md) before an
|
|
246
|
+
unattended job.
|
|
247
|
+
|
|
237
248
|
## Quick start
|
|
238
249
|
|
|
239
250
|
Create a local project on an NVIDIA host:
|
|
240
251
|
|
|
241
252
|
```bash
|
|
242
253
|
mkdir support-adapter && cd support-adapter
|
|
254
|
+
tt hardware
|
|
243
255
|
tt init \
|
|
244
256
|
--name "Support Adapter" \
|
|
245
257
|
--model Qwen/Qwen3.5-2B \
|
|
246
258
|
--profile spark
|
|
247
259
|
```
|
|
248
260
|
|
|
249
|
-
|
|
250
|
-
|
|
261
|
+
`tt hardware` reports which certified models this GPU can train, LoRA
|
|
262
|
+
fine-tune, or serve. The shell agent can run the same inventory via its
|
|
263
|
+
`examine_hardware` tool. Edit the generated `tunedtensor.json`, replacing both
|
|
264
|
+
placeholder examples, then preflight and run:
|
|
251
265
|
|
|
252
266
|
```bash
|
|
253
267
|
tt doctor tunedtensor.json
|