Function _is_quant_leaf

needle/model/quantize.py:150–153

(path, leaf)

def _is_quant_leaf(path, leaf):
    key = path[-1].key
    return ((key in ("kernel", "embedding") or key.startswith("mhc_phi"))
            and getattr(leaf, "ndim", 0) >= 2)


def _reduces_second_last(path):