load_state_dict

shrew::safetensors

Function load_state_dict 

Source 
pub fn load_state_dict<B: Backend>(
    path: impl AsRef<Path>,
    device: &B::Device,
) -> Result<HashMap<String, Tensor<B>>>
Expand description

Load parameters from a .safetensors file into a state-dict map.

Returns a HashMap<String, Tensor<B>> for flexible parameter loading.