From 85f0aaefe52414110fd93f3d050db236334ca090 Mon Sep 17 00:00:00 2001 From: Laurent Mazare Date: Wed, 22 Jan 2025 10:23:34 +0100 Subject: [PATCH] Add serde::serialize to activations. (#2732) --- candle-nn/src/activation.rs | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/candle-nn/src/activation.rs b/candle-nn/src/activation.rs index 772548a0..30f65de0 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]