Struct WeightLoader 

pub struct WeightLoader { /* private fields */ }

Weight loader for SSM models

Implementations

impl WeightLoader

pub fn new(config: WeightLoadConfig) -> Self

Create a new weight loader

pub fn load_safetensors<P: AsRef<Path>>( &self, path: P, varmap: &mut VarMap, ) -> CoreResult<()>

Load weights from a safetensors file

Note: This function uses varmap.load() which handles loading from safetensors format

pub fn save_safetensors<P: AsRef<Path>>( &self, path: P, varmap: &VarMap, ) -> CoreResult<()>

Save weights to a safetensors file

Note: This function uses varmap.save() which handles saving to safetensors format

pub fn load_pytorch_checkpoint<P: AsRef<Path>>( &self, _path: P, _varmap: &VarMap, ) -> CoreResult<()>

Load weights from PyTorch checkpoint

This is a placeholder for PyTorch checkpoint loading. Full implementation would require parsing PyTorch’s pickle format.