huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-17 02:58:50 +00:00
Code Issues Packages Projects Releases Wiki Activity

More pickle support. (#588)

Browse Source 
* More pickle support.

* Be more verbose.
This commit is contained in:
Laurent Mazare
2023-08-24 18:45:10 +01:00
committed by GitHub
parent e21c686cdc
commit 2cde0cb74b
2 changed files with 40 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
candle-core/examples/tensor-tools.rs
View File

@ -91,7 +91,7 @@ fn run_ls(file: &std::path::PathBuf, format: Option<Format>, verbose: bool) -> R
}
}
Format::Pth => {
let mut tensors = candle_core::pickle::read_pth_tensor_info(file)?;
let mut tensors = candle_core::pickle::read_pth_tensor_info(file, verbose)?;
tensors.sort_by(|a, b| a.name.cmp(&b.name));
for tensor_info in tensors.iter() {
println!(