diff options
| author | Dilshod Tadjibaev <939125+antimora@users.noreply.github.com> | 2024-02-06 14:17:33 -0600 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-02-06 21:17:33 +0100 |
| commit | b75e8945bc7c67106be6288b9f357efa8068e62e (patch) | |
| tree | e9918bcacc07281358f0ecfa66fa700278980cb1 /candle-core/examples/tensor-tools.rs | |
| parent | a90fc5ca5a486e988d39ea69ee3d3bb40a39c017 (diff) | |
| download | candle-b75e8945bc7c67106be6288b9f357efa8068e62e.tar.gz candle-b75e8945bc7c67106be6288b9f357efa8068e62e.tar.bz2 candle-b75e8945bc7c67106be6288b9f357efa8068e62e.zip | |
Enhance pickle to retrieve state_dict with a given key (#1671)
Diffstat (limited to 'candle-core/examples/tensor-tools.rs')
| -rw-r--r-- | candle-core/examples/tensor-tools.rs | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/candle-core/examples/tensor-tools.rs b/candle-core/examples/tensor-tools.rs index eb6ceb1c..1801ac58 100644 --- a/candle-core/examples/tensor-tools.rs +++ b/candle-core/examples/tensor-tools.rs @@ -196,7 +196,7 @@ fn run_ls( } } Format::Pth => { - let mut tensors = candle_core::pickle::read_pth_tensor_info(file, verbose)?; + let mut tensors = candle_core::pickle::read_pth_tensor_info(file, verbose, None)?; tensors.sort_by(|a, b| a.name.cmp(&b.name)); for tensor_info in tensors.iter() { println!( |