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

Added docstring for scale parameter

Browse Source
This commit is contained in:
Hazem 2023-12-23 00:22:44 +02:00
parent 0e5ae1d1bf
commit b9cbfbf3d7
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -23,6 +23,7 @@ class RoPE(Module):
implementation which is slightly less efficient. Default: ``False`` implementation which is slightly less efficient. Default: ``False``
base (float, optional): The base used to compute angular frequency for base (float, optional): The base used to compute angular frequency for
each dimension in the positional encodings. Default: ``10000`` each dimension in the positional encodings. Default: ``10000``
scale (float, optional): The scale used to scale the positions. Default: ``1.0``
""" """
def __init__( def __init__(