summaryrefslogtreecommitdiff
path: root/candle-core/src
diff options
context:
space:
mode:
authorlaurent <laurent.mazare@gmail.com>2023-07-03 08:44:00 +0100
committerlaurent <laurent.mazare@gmail.com>2023-07-03 08:44:00 +0100
commitfe2c07e36876b452e60d40197152b21f95e7d79f (patch)
tree3c72011e0aadbe27fc4ac9af70f91cf4941524d1 /candle-core/src
parentcf2789fb819049cb33a52d73b84a5810cc27cc97 (diff)
downloadcandle-fe2c07e36876b452e60d40197152b21f95e7d79f.tar.gz
candle-fe2c07e36876b452e60d40197152b21f95e7d79f.tar.bz2
candle-fe2c07e36876b452e60d40197152b21f95e7d79f.zip
Add the ST error.
Diffstat (limited to 'candle-core/src')
-rw-r--r--candle-core/src/error.rs4
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 },
}