Struct WeightLoader 

pub struct WeightLoader { /* private fields */ }

Builder for loading model weights with validation

Implementations

impl WeightLoader

pub fn new(loader: ModelLoader) -> Self

Create a new weight loader

pub fn model_type(self, model_type: ModelType) -> Self

Set the expected model type

pub fn strict(self, strict: bool) -> Self

Set whether to enforce strict loading (all weights must be present)

pub fn validate_weights(&self, required: &[&str]) -> ModelResult<()>

Validate that all required weights are present

pub fn loader(&self) -> &ModelLoader

Get the underlying loader

pub fn with_name_mapping(self, _mapping: HashMap<String, String>) -> Self

Create a name mapping from source format to target format

Example

let mapping = HashMap::from([
    ("backbone.layers.0.mixer.in_proj.weight", "layers.0.in_proj"),
    ("backbone.layers.0.mixer.A_log", "layers.0.ssm.log_a"),
]);
let mapped_loader = WeightLoader::new(loader).with_name_mapping(mapping);

pub fn print_weights(&self)

Print available weights and their shapes

This is useful for understanding what weights are available in the checkpoint

pub fn suggest_huggingface_mapping(&self) -> Vec<(String, String)>

Get suggested weight mappings for HuggingFace format

Returns a list of (hf_name, kizzasi_name) pairs that can be used to convert HuggingFace checkpoints to Kizzasi format