Skip to content

llmcompressor.modifiers.pruning.utils.pytorch.layer_mask

param_mask_name()

Name to use for mask buffer on a sparse layer

Source code in src/llmcompressor/modifiers/pruning/utils/pytorch/layer_mask.py 
13
14
15
16
17
def param_mask_name() -> str:
    """
    Name to use for mask buffer on a sparse layer
    """
    return "mask"