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

Doc error for default for scale in SinusoidalPositionalEncoding (#1174)

Browse Source
This commit is contained in:
Dominik Schlösser 2024-06-02 22:42:45 +02:00 committed by GitHub
parent 079882495d
commit 3576b547c5
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
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

@ -67,7 +67,7 @@ class SinusoidalPositionalEncoding(Module):
max_freq (float, optional): The maximum frequency expected. Default: max_freq (float, optional): The maximum frequency expected. Default:
``1``. ``1``.
scale (float, optional): A multiplicative scale for the embeddings. scale (float, optional): A multiplicative scale for the embeddings.
Default: ``sqrt(dims//2)``. Default: ``sqrt(2/dims)``.
cos_first (bool, optional): If ``True`` embed using ``[cos(x); sin(x)]`` cos_first (bool, optional): If ``True`` embed using ``[cos(x); sin(x)]``
instead of the reverse. Default: ``False``. instead of the reverse. Default: ``False``.
full_turns (bool, optional): If ``True`` multiply the frequencies with full_turns (bool, optional): If ``True`` multiply the frequencies with