diff options
| -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 { |