Fix arange call with latest MLX.

lucasnewman · Jan 9, 2025 · e0bae96 · e0bae96
1 parent 9541e7c
commit e0bae96
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/f5_tts_mlx/utils.py b/f5_tts_mlx/utils.py
@@ -40,7 +40,7 @@ def lens_to_mask(
     length: int | None = None,
 ) -> mx.array:  # Bool['b n']
     if not exists(length):
-        length = t.max()
+        length = t.max().item()
 
     seq = mx.arange(length)
     return einx.less("n, b -> b n", seq, t)

diff --git a/pyproject.toml b/pyproject.toml
@@ -6,7 +6,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "f5-tts-mlx"
-version = "0.2.3"
+version = "0.2.4"
 authors = [{name = "Lucas Newman", email = "lucasnewman@me.com"}]
 license = {text = "MIT"}
 description = "F5-TTS - MLX"