transformer_lens.conversion_utils.helpers.merge_quantiziation_fields module¶

Merge quantization fields helper.

This module contains helper functions for merging quantization fields.

transformer_lens.conversion_utils.helpers.merge_quantiziation_fields.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).