huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-18 19:47:12 +00:00
Code Issues Packages Projects Releases Wiki Activity

Added comment about offsets.

Browse Source
This commit is contained in:
Nicolas Patry
2023-07-27 20:11:57 +02:00
parent 952eca6b54
commit 8435a99edd
1 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
candle-nn/src/var_builder.rs
View File

@ -187,6 +187,9 @@ impl<'a> VarBuilder<'a> {
let start = rank * block_size; let start = rank * block_size;
let stop = (rank + 1) * block_size; let stop = (rank + 1) * block_size;
// Everything is expressed in tensor dimension
// bytes offsets is handled automatically for safetensors.
let iterator = if dim == 0 { let iterator = if dim == 0 {
view.slice(start..stop).map_err(|_| Error::Msg(format!("Cannot slice tensor {tensor_name} ({shape:?} along dim {dim} with {start}..{stop}")))? view.slice(start..stop).map_err(|_| Error::Msg(format!("Cannot slice tensor {tensor_name} ({shape:?} along dim {dim} with {start}..{stop}")))?
} else if dim == 1 { } else if dim == 1 {