From fe2c07e36876b452e60d40197152b21f95e7d79f Mon Sep 17 00:00:00 2001 From: laurent Date: Mon, 3 Jul 2023 08:44:00 +0100 Subject: [PATCH] Add the ST error. --- candle-core/src/error.rs | 4 ++++ 1 file changed, 4 insertions(+) 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 }, }