transformer_lens.model_bridge.get_params_util module

Utility function for getting model parameters in TransformerLens format.

transformer_lens.model_bridge.get_params_util.get_bridge_params(bridge) → Dict[str, Tensor]

Model parameters in SVDInterpreter format. Skips attn keys for non-attention layers.