transformer_lens.conversion_utils.helpers package

Submodules

• transformer_lens.conversion_utils.helpers.find_property module
  • find_property()
• transformer_lens.conversion_utils.helpers.merge_quantiziation_fields module
  • merge_quantization_fields()

Module contents

Helper functions for architecture adapter conversion.

transformer_lens.conversion_utils.helpers.find_property(needle: str, haystack: object | dict)

transformer_lens.conversion_utils.helpers.merge_quantization_fields(field_set: Any, quantization_fields: dict[str, Any]) → Any

Merge quantization fields into a field set.

Parameters:

• field_set – The field set to merge into.

• quantization_fields – The quantization fields to merge.

Returns:

The merged field set (same object, modified in-place).