transformer_lens.model_bridge.supported_architectures.lfm2 module¶

Lfm2 architecture adapter.

class transformer_lens.model_bridge.supported_architectures.lfm2.Lfm2ArchitectureAdapter(cfg: Any)¶

Bases: ArchitectureAdapter

Architecture adapter for Lfm2 models.

__init__(cfg: Any) → None¶: Initialize the Lfm2 architecture adapter.

setup_component_testing(hf_model: Any, bridge_model: Any = None) → None¶: Set up model-specific references for component testing.