Add the ST error.

This commit is contained in:
laurent
2023-07-03 08:44:00 +01:00
parent cf2789fb81
commit fe2c07e368

View File

@ -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 },
}