classify_key

Function classify_key 

Source
pub fn classify_key(
    key: &str,
    cfg: &ModelConfig,
) -> Result<(usize, LoraTarget, AdapterAb)>
Expand description

PEFT key → (layer, module, A|B). Every unsupported shape is a NAMED hard rejection — never a skip. Prefix-agnostic on purpose: the Holo base checkpoint keys are model.language_model.layers.{i}.* (weight_prefix auto-detected server-side), but a PEFT trainer wrapping the text trunk emits model.layers.{i}.*; both carry the layer index right after “.layers.”.