Skip to main content

load_state_dict

Function load_state_dict 

Source
pub fn load_state_dict(
    path: &Path,
) -> Result<HashMap<String, Tensor>, ModelError>
Expand description

Load all tensors from a SafeTensors file into a name-to-tensor map.

All tensors are converted to F32.