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

Retrieve more information from PyTorch checkpoints. (#515)

Browse Source 
* Retrieve more information from PyTorch checkpoints.

* Add enough support to load dino-v2 backbone weights.
This commit is contained in:
Laurent Mazare
2023-08-19 15:05:34 +01:00
committed by GitHub
parent f861a9df6e
commit 607ffb9f1e
3 changed files with 75 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -88,9 +88,15 @@ fn run_ls(file: &std::path::PathBuf, format: Option<Format>) -> Result<()> {
}
Format::PyTorch => {
let mut tensors = candle_core::pickle::read_pth_tensor_info(file)?;
tensors.sort_by(|a, b| a.0.cmp(&b.0));
for (name, dtype, shape) in tensors.iter() {
println!("{name}: [{shape:?}; {dtype:?}]")
tensors.sort_by(|a, b| a.name.cmp(&b.name));
for tensor_info in tensors.iter() {
println!(
"{}: [{:?}; {:?}] {:?}",
tensor_info.name,
tensor_info.layout.shape(),
tensor_info.dtype,
tensor_info.path,
)
}
}
Format::Pickle => {