Bump extension MLX version (#1350)

* Bump extension MLX version

* fix some docs nits
This commit is contained in:
Awni Hannun
2024-08-23 12:38:34 -07:00
committed by GitHub
parent 684e11c664
commit 3b4d5484c7
4 changed files with 6 additions and 4 deletions

View File

@@ -102,11 +102,12 @@ void init_fast(nb::module_& parent_module) {
implementation which rotates consecutive dimensions.
base (float, optional): The base used to compute angular frequency for
each dimension in the positional encodings. Exactly one of ``base`` and
``freqs`` must be ``None``.
``freqs`` must be ``None``.
scale (float): The scale used to scale the positions.
offset (int): The position offset to start at.
freqs (array, optional): Optional frequencies to use with RoPE.
If set, the ``base`` parameter must be ``None``. ``Default: None``.
If set, the ``base`` parameter must be ``None``. Default: ``None``.
Returns:
array: The output array.
)pbdoc");