rm unrelated derive

coreylowman · Feb 1, 2024 · de55567 · de55567
1 parent b52932c
commit de55567
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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]