Added docstring for scale parameter

2025-08-21 02:46:40 +08:00 · 2023-12-23 00:22:44 +02:00 · 2023-12-23 00:22:44 +02:00 · b9cbfbf3d7
commit b9cbfbf3d7
parent 0e5ae1d1bf
1 changed files with 1 additions and 0 deletions
--- a/python/mlx/nn/layers/positional_encoding.py
+++ b/python/mlx/nn/layers/positional_encoding.py
@ -23,6 +23,7 @@ class RoPE(Module):
            implementation which is slightly less efficient. Default: ``False``
        base (float, optional): The base used to compute angular frequency for
            each dimension in the positional encodings. Default: ``10000``
+        scale (float, optional): The scale used to scale the positions. Default: ``1.0``
    """

    def __init__(