dnrops
/
candle
mirror of https://github.com/huggingface/candle.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Add serde::serialize to activations. (#2732)

This commit is contained in:
Laurent Mazare 2025-01-22 10:23:34 +01:00 committed by GitHub
parent e4c3a71f11
commit 85f0aaefe5
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
candle-nn/src/activation.rs
View File

@ -1,9 +1,8 @@
//! Activation Functions //! Activation Functions
//! //!
use candle::{Result, Tensor}; 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")] #[serde(rename_all = "lowercase")]
pub enum Activation { pub enum Activation {
#[default] #[default]