diff options
| -rw-r--r-- | candle-core/src/error.rs | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/candle-core/src/error.rs b/candle-core/src/error.rs index 341fc151..71fd21de 100644 --- a/candle-core/src/error.rs +++ b/candle-core/src/error.rs @@ -110,6 +110,10 @@ pub enum Error { #[error(transparent)] Io(#[from] std::io::Error), + /// SafeTensor error. + #[error(transparent)] + SafeTensor(#[from] safetensors::SafeTensorError), + #[error("cannot broadcast {src_shape:?} to {dst_shape:?}")] BroadcastIncompatibleShapes { src_shape: Shape, dst_shape: Shape }, } |