Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
5 changes: 5 additions & 0 deletions STATUS.md
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,11 @@

As of: 2026-06-07

> **Planned-run truth:** This file is a historical capability snapshot. The
> machine-readable source of truth for the next training run is
> `configs/runs/current_run.yaml`; null values listed there are launch blockers,
> not values to infer from this document.

This is the current short-form truth for the repo. If old phase plans,
April/May status states, or specs contradict it, this file takes precedence first,
then the reports from 2026-05-29, then the respective working docs.
Expand Down
122 changes: 122 additions & 0 deletions configs/runs/current_run.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,122 @@
schema_version: 1
status: draft_waiting_for_data_measurements
measured_at: "2026-07-10"
purpose: german_primary_40b_continued_pretraining

# This file describes the next run, not historical experiments. Null values are
# launch blockers and must be replaced by measured values rather than guesses.
code:
repository: AuraIis/Auralis
required_ref: main
resolved_commit_sha: null
server_checkout:
required_mode: clean_repo_checkout
forbidden_training_tree: NEWGPT/v2data
working_tree_must_be_clean: true
container_source_must_match_resolved_commit: true
record_at_launch:
- git_rev_parse_head
- git_status_porcelain
- container_source_tree_hash

base_checkpoint:
path: corpus20b_codeheavy/step_60000
tokens_seen: 3930000000
validation_loss: 1.843
gate_score: 0.14
creation_environment_reported:
pytorch: "2.12"
cuda: cu130

tokenizer:
immutable_for_run: true
measured_hash: a24fbea439bc8b78
measured_hash_scope: sha256_prefix_16
full_sha256: null

runtime_environment:
python: null
pytorch: 2.7.0+cu128
cuda_wheel: cu128
gpu_architecture_reported: blackwell
required_environment:
TRITON_OVERRIDE_ARCH: sm89
compatibility_note: >-
The base checkpoint was created under the reported torch 2.12/cu130 environment.
The current server uses torch 2.7.0+cu128 and requires TRITON_OVERRIDE_ARCH=sm89
to avoid the observed Mamba Triton PassManager failure.

training:
planned_total_tokens: 40000000000
architecture_frozen_until_data_ab: true
mix_percent:
german: 65
english: 15
code: 12
math_science: 8

data:
german_measured_tokens: 7460000000
german_components:
- id: de_curated
path: curated_40b/german.bin
tokens: 3240000000
sha256: null
- id: german_fresh
path: null
tokens: 2670000000
sha256: null
- id: german_commons
path: null
tokens: 1550000000
sha256: null
forbidden_inputs:
- path: corpus20b/de_curated.bin
observed_size_bytes: 25
reason: placeholder_stub_not_training_data
deduplication:
implementation: scripts/data/dedup_de_fresh.py
mode: cross_dataset_against_existing_reference
reference_manifest: null
defaults:
normalized_exact_hash: sha1
minhash_num_perm: 64
minhash_threshold: 0.85
word_shingle_size: 5

fineweb2_hq_de_pilot:
status: retention_measurement_running
gross_tokens: 28500000000
storage_with_embeddings_gb: 663
estimated_text_only_gb: 155
dedup_reference: german_fresh
measured_index_rate_docs_per_second: 1230
dedup_report: null
promotion_rule: >-
Scale only after retained tokens, exact/near drop counts, quality probes and
cross-source dedup retention are recorded. Otherwise move budget to
FinePDFs-Edu or HPLT2-DE.

verified_checks:
single_gpu_resume:
status: passed
resume_step: 20
compared_steps: [25, 30]
loss_and_learning_rate: bit_identical
two_gpu_ddp:
status: pending_hardware
sparse_attention_flash_variant:
state_dict: identical
semantics: intentionally_different
standard_global_tokens: 32
standard_window: 1024
flash_global_tokens: 0
flash_window: 2048
parity_test: pending

launch_blockers:
- resolved_commit_sha
- tokenizer.full_sha256
- data.german_components.sha256
- data.deduplication.reference_manifest
- fineweb2_hq_de_pilot.dedup_report
52 changes: 52 additions & 0 deletions tests/config/test_current_run.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,52 @@
from pathlib import Path

ROOT = Path(__file__).resolve().parents[2]
CURRENT_RUN = ROOT / "configs" / "runs" / "current_run.yaml"


def load_current_run() -> dict:
import yaml

return yaml.safe_load(CURRENT_RUN.read_text(encoding="utf-8"))


def test_current_run_mix_is_complete_and_german_primary() -> None:
run = load_current_run()
mix = run["training"]["mix_percent"]

assert sum(mix.values()) == 100
assert mix == {"german": 65, "english": 15, "code": 12, "math_science": 8}


def test_current_run_cannot_silently_use_stale_server_tree_or_stub() -> None:
run = load_current_run()

assert run["code"]["server_checkout"]["required_mode"] == "clean_repo_checkout"
assert run["code"]["server_checkout"]["forbidden_training_tree"] == "NEWGPT/v2data"
assert run["data"]["forbidden_inputs"] == [
{
"path": "corpus20b/de_curated.bin",
"observed_size_bytes": 25,
"reason": "placeholder_stub_not_training_data",
}
]


def test_current_run_keeps_unmeasured_provenance_as_launch_blockers() -> None:
run = load_current_run()

assert run["status"] == "draft_waiting_for_data_measurements"
assert run["code"]["resolved_commit_sha"] is None
assert run["tokenizer"]["full_sha256"] is None
assert run["data"]["deduplication"]["reference_manifest"] is None
assert run["fineweb2_hq_de_pilot"]["dedup_report"] is None
assert "resolved_commit_sha" in run["launch_blockers"]


def test_current_run_records_required_blackwell_compatibility_override() -> None:
run = load_current_run()

assert run["runtime_environment"]["pytorch"] == "2.7.0+cu128"
assert run["runtime_environment"]["required_environment"] == {
"TRITON_OVERRIDE_ARCH": "sm89"
}
Loading