transformer_lens.model_bridge.supported_architectures.gemma1 module

Gemma1 architecture adapter.

class transformer_lens.model_bridge.supported_architectures.gemma1.Gemma1ArchitectureAdapter(cfg: Any)

Bases: ArchitectureAdapter

Architecture adapter for Gemma1 models.

__init__(cfg: Any) → None

Initialize the Gemma1 architecture adapter.