From b9cbfbf3d746c81577058e31b8749347edb7d269 Mon Sep 17 00:00:00 2001 From: Hazem Date: Sat, 23 Dec 2023 00:22:44 +0200 Subject: [PATCH] Added docstring for scale parameter --- python/mlx/nn/layers/positional_encoding.py | 1 + 1 file changed, 1 insertion(+) diff --git a/python/mlx/nn/layers/positional_encoding.py b/python/mlx/nn/layers/positional_encoding.py index 4852df546..32d5e2c88 100644 --- 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__(