transformer_lens.model_bridge.supported_architectures.gptj module

GPTJ architecture adapter.

class transformer_lens.model_bridge.supported_architectures.gptj.GptjArchitectureAdapter(cfg: Any)

Bases: ArchitectureAdapter

Architecture adapter for GPTJ models.

__init__(cfg: Any) → None

Initialize the GPTJ architecture adapter.