transformer_lens.model_bridge.sources.inspect.masks module

Torch-free validation of the single-sequence Inspect padding-mask wire format.

transformer_lens.model_bridge.sources.inspect.masks.normalize_attention_mask(mask: Any, n_tokens: int) → list[int]

Accept a binary flat wire mask or a single-row public padding mask.