pub fn prefill_head(spec: &ModelSpec, dir: &str, chunk: usize) -> Authored
A prefill chunk’s head alone: the last layer’s stacked output in, the next token after the chunk’s last out.