Skip to content

llmcompressor.core.model_layer

ModelParameterizedLayer dataclass

A dataclass for holding a parameter and its layer

Parameters:

Name Type Description Default
layer_name str

the name of the layer

required
layer Any

the layer object

required
param_name str

the name of the parameter

required
param Any

the parameter object

required
Source code in src/llmcompressor/core/model_layer.py
@dataclass
class ModelParameterizedLayer:
    """
    A dataclass for holding a parameter and its layer

    :param layer_name: the name of the layer
    :param layer: the layer object
    :param param_name: the name of the parameter
    :param param: the parameter object
    """

    layer_name: str
    layer: Any
    param_name: str
    param: Any