diff --git a/python/mlx/nn/layers/activations.py b/python/mlx/nn/layers/activations.py index ba565e02d..88ff3476a 100644 --- a/python/mlx/nn/layers/activations.py +++ b/python/mlx/nn/layers/activations.py @@ -315,7 +315,7 @@ class GLU(Module): return glu(x=x, axis=self.axis) -@_make_activation_module(mx.sigmoid) +@_make_activation_module(sigmoid) class Sigmoid(Module): r"""Applies the sigmoid function, element-wise.