transformer_lens.model_bridge.supported_architectures.bart module

BART architecture adapter.

class transformer_lens.model_bridge.supported_architectures.bart.BartArchitectureAdapter(cfg: Any)

Bases: ArchitectureAdapter

Architecture adapter for BartForConditionalGeneration models.

__init__(cfg: Any) → None

Initialize the BART architecture adapter.