Function clear_act_buffer

lib/model_wrapper.py:137–140

(act_model)

def clear_act_buffer(act_model):
    for name, module in act_model.named_modules():
        if isinstance(module, ActLinear):
            module.clear_act_buffer()


def prune_wanda_v2(