zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-08-22 21:16:47 +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
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats 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