Struct ModelLoader 

pub struct ModelLoader { /* private fields */ }

Weight loader for safetensors format

Implementations

impl ModelLoader

pub fn new<P: AsRef<Path>>(path: P) -> ModelResult<Self>

Load a safetensors file from disk

pub fn from_bytes(data: Vec<u8>) -> ModelResult<Self>

Load a safetensors from bytes

pub fn list_tensors(&self) -> Vec<String>

List all available tensor names in the file

pub fn tensor_info(&self, name: &str) -> Option<TensorInfo>

Get metadata about a specific tensor

pub fn load_array1(&self, name: &str) -> ModelResult<Array1<f32>>

Load a 1D tensor (Array1)

pub fn load_array2(&self, name: &str) -> ModelResult<Array2<f32>>

Load a 2D tensor (Array2)

pub fn load_array(&self, name: &str) -> ModelResult<ArrayD<f32>>

Load a tensor of arbitrary dimension

pub fn load_array3(&self, name: &str) -> ModelResult<Vec<Vec<Vec<f32>>>>

Load a 3D tensor as Vec<Vec<Vec>>

This is useful for convolution weights [out_channels, in_channels, kernel_size]

pub fn has_tensor(&self, name: &str) -> bool

Check if a tensor exists

pub fn load_all(&self) -> ModelResult<HashMap<String, ArrayD<f32>>>

Load all tensors into a HashMap

pub fn print_summary(&self)

Print a summary of all tensors in the file

This is useful for inspecting checkpoint files and understanding their structure

pub fn get_size_stats(&self) -> HashMap<String, usize>

Get statistics about tensor sizes

pub fn search_tensors(&self, pattern: &str) -> Vec<String>

Search for tensors matching a pattern

Example

// Find all conv weights
let conv_tensors = loader.search_tensors("conv.weight");