You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
We reimplemented BLT architecture is concentrated in four files:
22
22
@@ -42,7 +42,7 @@ We reimplemented BLT architecture is concentrated in four files:
42
42
We intentionally left distributed/runtime infrastructure mostly intact and
43
43
focused on the model code that defines the architecture.
44
44
45
-
## Patching And Representations
45
+
## Patching, representations
46
46
47
47
The patch-length embedding experiment is implemented in `bytelatent/model/blt.py`.
48
48
After the local encoder builds patch states, the model can convert `patch_lengths`
@@ -51,7 +51,7 @@ patch states before the global transformer. This is controlled by
51
51
`use_patch_length_sinusoidal_embedding` and enabled in
52
52
`bytelatent/configs/patch_len_embeddings.yaml`.
53
53
54
-
## GPT-2 Baseline
54
+
## GPT-2 baseline
55
55
56
56
`apps/main/train_distilgpt2.py` is the HuggingFace GPT-2 baseline trainer we made to compare with the BLT architecture. It builds a `GPT2LMHeadModel` from config, streams FineWeb-Edu text, tokenizes with
57
57
the `distilgpt2` BPE tokenizer, trains with AdamW and cosine LR, logs BPB using
@@ -73,7 +73,7 @@ Older GPT-2-shaped configs also exist in `apps/main/configs/distilgpt2_83m*.yaml
73
73
Those follow the repo's BLT training-config shape and were useful during setup,
74
74
but the HF trainer path above is the original baseline path.
75
75
76
-
## Training Configs
76
+
## Training configs
77
77
78
78
The BLT configs inherit from each other using the `config:` key.
79
79
@@ -98,7 +98,7 @@ The BLT configs inherit from each other using the `config:` key.
98
98
This uses `use_patch_length_sinusoidal_embedding` so the global transformer sees
99
99
patch length as an extra signal.
100
100
101
-
## Experiment Summary
101
+
## Experiment summary
102
102
103
103
| Variant | Config | N-gram | Params | BPB |
104
104
| --- | --- | --- | --- | --- |
@@ -115,7 +115,7 @@ Shrinking either local side too aggressively creates a bottleneck. A concentrate
115
115
n-gram lengths with smaller tables, and sinusoidal patch-length embeddings gave
116
116
us a direct patch-representation ablation.
117
117
118
-
## Environment Setup
118
+
## Env setups
119
119
120
120
To run experiments on Prime Intellect H100s, we needed a repeatable way to turn
121
121
a fresh Ubuntu GPU box into a BLT training machine. `setup/create_env_uv.sh`
0 commit comments