Skip to content

Commit

Permalink
rm unrelated derive
Browse files Browse the repository at this point in the history
  • Loading branch information
swfsql committed Feb 1, 2024
1 parent b52932c commit de55567
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion dfdx/src/nn/layers/layer_rms_norm1d.rs
Original file line number Diff line number Diff line change
Expand Up @@ -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]
Expand Down

0 comments on commit de55567

Please sign in to comment.