diff --git a/candle-nn/src/activation.rs b/candle-nn/src/activation.rs index 772548a01a..30f65de08a 100644 --- a/candle-nn/src/activation.rs +++ b/candle-nn/src/activation.rs @@ -1,9 +1,8 @@ //! Activation Functions //! use candle::{Result, Tensor}; -use serde::Deserialize; -#[derive(Debug, Clone, Copy, PartialEq, Deserialize, Default)] +#[derive(Debug, Clone, Copy, PartialEq, serde::Deserialize, serde::Serialize, Default)] #[serde(rename_all = "lowercase")] pub enum Activation { #[default]