load

Function load 

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

Load named tensors from a .safetensors file.

use shrew::safetensors;
use shrew::prelude::*;

let tensors = safetensors::load::<CpuBackend>("model.safetensors", &CpuDevice).unwrap();
for (name, tensor) in &tensors {
    println!("{name}: {:?}", tensor.dims());
}