From fdcd9432e3c101cc759a784907568c76c155852e Mon Sep 17 00:00:00 2001
From: Thiago Machado <swfsql@gmail.com>
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<M: Dim, E: Dtype, D: Device<E>> BuildOnDevice<E, D> 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<M: Dim, Elem: Dtype, Dev: Device<Elem>> {
     #[param]