Skip to content

llmcompressor.modifiers.pruning.utils.pytorch.layer_mask

Functions:

param_mask_name 

param_mask_name() -> str

Name to use for mask buffer on a sparse layer

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