AbstractPhil commited on
Commit
34d7ec4
·
verified ·
1 Parent(s): 943c0c2

mirror 0.7.4: mini-beatrix-2 hub_chunk 1024 (AlephLLM@cb7cb86)

Browse files
Files changed (1) hide show
  1. presets.py +6 -1
presets.py CHANGED
@@ -163,7 +163,12 @@ PRESETS: dict[str, Preset] = {
163
  hub_layers=tuple(range(32)),
164
  hub_K=256, hub_D=256, hub_const=16,
165
  bank_experts=6, bank_ff=1024,
166
- head_K=256, head_D=256, hub_chunk=256,
 
 
 
 
 
167
  hub_ckpt=2),
168
  train=TrainConfig(micro_batch=4, grad_accum=16,
169
  governor="minsep", governor_theta=45.0),
 
163
  hub_layers=tuple(range(32)),
164
  hub_K=256, hub_D=256, hub_const=16,
165
  bank_experts=6, bank_ff=1024,
166
+ # chunk 1024 MEASURED on the mission card (C2e,
167
+ # Blackwell 2026-08-26): 72.2 vs 83.2 ms/layer
168
+ # fwd+bwd at chunk 256, peak 39.4 -> 26.6 GB.
169
+ # S/P traffic ~ 1/C, att work ~ C; config-only,
170
+ # checkpoint-compatible, exactness C-independent.
171
+ head_K=256, head_D=256, hub_chunk=1024,
172
  hub_ckpt=2),
173
  train=TrainConfig(micro_batch=4, grad_accum=16,
174
  governor="minsep", governor_theta=45.0),