Commit Graph

2 Commits

Author SHA1 Message Date
617da29a5b fix: assert topk_idx is not None in CSA layers — no silent fallback to SWA-only
The indexer silently returning None caused CSA layers to attend over only the
SWA window (128 tokens), not the compressed sparse KV. This went undetected
because the model still produced plausible output at short context. The assert
makes any future indexer regression immediately visible.
2026-06-02 06:14:23 +00:00
938e9079ce probe: indexer and compressor weight shapes from checkpoint 2026-06-02 04:36:35 +00:00