diff options
Diffstat (limited to 'candle-core/examples/tensor-tools.rs')
| -rw-r--r-- | candle-core/examples/tensor-tools.rs | 3 |
1 files changed, 1 insertions, 2 deletions
diff --git a/candle-core/examples/tensor-tools.rs b/candle-core/examples/tensor-tools.rs index c0d5a334..3982f2c3 100644 --- a/candle-core/examples/tensor-tools.rs +++ b/candle-core/examples/tensor-tools.rs @@ -150,8 +150,7 @@ fn run_ls(file: &std::path::PathBuf, format: Option<Format>, verbose: bool) -> R } } Format::Safetensors => { - let tensors = unsafe { candle_core::safetensors::MmapedFile::new(file)? }; - let tensors = tensors.deserialize()?; + let tensors = unsafe { candle_core::safetensors::MmapedSafetensors::new(file)? }; let mut tensors = tensors.tensors(); tensors.sort_by(|a, b| a.0.cmp(&b.0)); for (name, view) in tensors.iter() { |