transformer_lens.conversion_utils.conversion_steps.tensor_conversion_set module

Tensor conversion set.

class transformer_lens.conversion_utils.conversion_steps.tensor_conversion_set.TensorConversionSet(fields: dict[str, Any])

Bases: BaseTensorConversion

get_component(model: Any, name: str) → Any

Get a component from the model using the field mapping.

Parameters:

  • model – The model to get the component from.

  • name – The name of the component to get.

Returns:

The requested component.

get_conversion_action(field: str) → BaseTensorConversion
handle_conversion(input_value: Any, *full_context: Any) → dict[str, Any]
process_conversion(input_value: Any, remote_field: str, conversion: BaseTensorConversion, *full_context: Any) → Any
process_conversion_action(input_value: Any, conversion_details: Any, *full_context: Any) → Any