transformer_lens.model_bridge.supported_architectures.gemma2 module

Gemma2 architecture adapter.

class transformer_lens.model_bridge.supported_architectures.gemma2.Gemma2ArchitectureAdapter(cfg: Any)

Bases: ArchitectureAdapter

Architecture adapter for Gemma2 models.

__init__(cfg: Any) → None

Initialize the Gemma2 architecture adapter.