transformer_lens.factories.architecture_adapter_factory module

Architecture adapter factory.

This module provides a factory for creating architecture adapters.

class transformer_lens.factories.architecture_adapter_factory.ArchitectureAdapterFactory

Bases: object

Factory for creating architecture adapters.

classmethod select_architecture_adapter(cfg: TransformerBridgeConfig) → ArchitectureAdapter

Select the appropriate architecture adapter for the given config.

Parameters:

cfg – The TransformerBridgeConfig to select the adapter for.

Returns:

The selected architecture adapter.

Raises:

ValueError – If no adapter is found for the given config.