zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-10-08 00:28:10 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix style check (#395)

Browse Source
This commit is contained in:
Angelos Katharopoulos
2024-01-07 05:54:58 -08:00
committed by GitHub
parent 48f6ca8c3a
commit 6ea6b4258d
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
python/mlx/nn/layers/positional_encoding.py
View File

@@ -227,4 +227,4 @@ class ALiBi(Module):
) )
if mask is not None: if mask is not None:
alibi_mask = alibi_mask + mask alibi_mask = alibi_mask + mask
return attention_scores + alibi_mask return attention_scores + alibi_mask