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());
}