diff options
| author | Nicolas Patry <patry.nicolas@protonmail.com> | 2023-07-27 20:11:57 +0200 |
|---|---|---|
| committer | Nicolas Patry <patry.nicolas@protonmail.com> | 2023-07-27 20:11:57 +0200 |
| commit | 8435a99edd6f5aa7bc86d0ffdaf23e322a93f626 (patch) | |
| tree | e30111fd7ac74aaf365391d234c8bd160c100ac6 | |
| parent | 952eca6b540078b1f30b58d9eb930f8e32d903cb (diff) | |
| download | candle-8435a99edd6f5aa7bc86d0ffdaf23e322a93f626.tar.gz candle-8435a99edd6f5aa7bc86d0ffdaf23e322a93f626.tar.bz2 candle-8435a99edd6f5aa7bc86d0ffdaf23e322a93f626.zip | |
Added comment about offsets.
| -rw-r--r-- | candle-nn/src/var_builder.rs | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/candle-nn/src/var_builder.rs b/candle-nn/src/var_builder.rs index 3133f210..5c222bf6 100644 --- a/candle-nn/src/var_builder.rs +++ b/candle-nn/src/var_builder.rs @@ -187,6 +187,9 @@ impl<'a> VarBuilder<'a> { let start = rank * 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 { 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 { |