From 6ea6b4258d45caee13dd1da5879cc6852166bbc4 Mon Sep 17 00:00:00 2001 From: Angelos Katharopoulos Date: Sun, 7 Jan 2024 05:54:58 -0800 Subject: [PATCH] Fix style check (#395) --- python/mlx/nn/layers/positional_encoding.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/mlx/nn/layers/positional_encoding.py b/python/mlx/nn/layers/positional_encoding.py index 7ed4a19d7..1c586693f 100644 --- a/python/mlx/nn/layers/positional_encoding.py +++ b/python/mlx/nn/layers/positional_encoding.py @@ -227,4 +227,4 @@ class ALiBi(Module): ) if mask is not None: alibi_mask = alibi_mask + mask - return attention_scores + alibi_mask \ No newline at end of file + return attention_scores + alibi_mask