Add serde::serialize to activations.

huggingface · Jan 22, 2025 · 469255e · 469255e
1 parent e4c3a71
commit 469255e
Showing 1 changed file with 1 addition and 2 deletions.
diff --git 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]