From 5be5daa6ef10688b78d5ecc22dda0d3be13fa23f Mon Sep 17 00:00:00 2001 From: Cheng Date: Tue, 14 May 2024 21:25:57 +0800 Subject: [PATCH] Use compiled function in Sigmoid module (#1116) --- python/mlx/nn/layers/activations.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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.