From 8435a99edd6f5aa7bc86d0ffdaf23e322a93f626 Mon Sep 17 00:00:00 2001 From: Nicolas Patry Date: Thu, 27 Jul 2023 20:11:57 +0200 Subject: [PATCH] Added comment about offsets. --- candle-nn/src/var_builder.rs | 3 +++ 1 file changed, 3 insertions(+) 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 {