transformer_lens.model_bridge.supported_architectures.gpt2_lm_head_custom module

GPT-2 LM Head Custom architecture adapter.

class transformer_lens.model_bridge.supported_architectures.gpt2_lm_head_custom.Gpt2LmHeadCustomArchitectureAdapter(cfg: Any)

Bases: ArchitectureAdapter

Architecture adapter for GPT-2 LM Head Custom models.

__init__(cfg: Any) → None

Initialize the GPT-2 LM Head Custom architecture adapter.