From de5556737ec560990c4278418f1f31f16d532d21 Mon Sep 17 00:00:00 2001 From: Thiago Machado Date: Thu, 1 Feb 2024 11:25:07 -0500 Subject: [PATCH] rm unrelated derive --- dfdx/src/nn/layers/layer_rms_norm1d.rs | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/dfdx/src/nn/layers/layer_rms_norm1d.rs b/dfdx/src/nn/layers/layer_rms_norm1d.rs index 17143aef..a62fffb9 100644 --- a/dfdx/src/nn/layers/layer_rms_norm1d.rs +++ b/dfdx/src/nn/layers/layer_rms_norm1d.rs @@ -38,7 +38,7 @@ impl> BuildOnDevice for LayerRMSNorm1DConfi } /// See [LayerRMSNorm1DConfig] -#[derive(Clone, Debug, UpdateParams, ZeroGrads, WithGrads)] +#[derive(Clone, Debug, UpdateParams, ZeroGrads)] #[cfg_attr(feature = "safetensors", derive(SaveSafeTensors, LoadSafeTensors))] pub struct LayerRMSNorm1D> { #[param]