Skip to main content

map_tensor_name

Function map_tensor_name 

Source
pub fn map_tensor_name(hf_name: &str) -> Option<String>
Expand description

Map a FlowState HuggingFace tensor name to the GGUF canonical name. Returns None for tensors unused at inference time (caller skips them).

HF prefix in safetensors: (none — model is saved as FlowStateForPrediction, which wraps FlowStateModel in self.model, but the checkpoint stores FlowStateModel weights directly without the .model. prefix.)

Verified tensor names from model.safetensors header: embed.embed.weight / embed.embed.bias encoder.layers.{N}.ssm.{log_Lambda_real, Lambda_imag, B_tilde_r, B_tilde_i, C_tilde_r, C_tilde_i, D, log_Delta} encoder.layers.{N}.out.{weight, bias} encoder.layers.{N}.norm.{weight, bias} decoder.lin.{weight, bias}