Merge pull request #102 from kyutai-labs/mlx-longer-conversations

Allow longer conversations with mlx.
kyutai-labs · Sep 22, 2024 · b20815e · b20815e
2 parents c7439c1 + d6b48c2
commit b20815e
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 4 deletions.
diff --git a/moshi_mlx/moshi_mlx/local.py b/moshi_mlx/moshi_mlx/local.py
@@ -256,7 +256,7 @@ def main():
     parser.add_argument("--moshi-weight", type=str)
     parser.add_argument("--mimi-weight", type=str)
     parser.add_argument("-q", "--quantized", type=int, choices=[4, 8])
-    parser.add_argument("--steps", default=2500, type=int)
+    parser.add_argument("--steps", default=4000, type=int)
     parser.add_argument("--hf-repo", type=str, default=None)
 
     args = parser.parse_args()

diff --git a/moshi_mlx/moshi_mlx/local_web.py b/moshi_mlx/moshi_mlx/local_web.py
@@ -358,7 +358,7 @@ def main():
     parser.add_argument("--moshi-weight", type=str)
     parser.add_argument("--mimi-weight", type=str)
     parser.add_argument("-q", "--quantized", type=int, choices=[4, 8])
-    parser.add_argument("--steps", default=2500, type=int)
+    parser.add_argument("--steps", default=4000, type=int)
     parser.add_argument("--hf-repo", type=str)
     parser.add_argument("--static", type=str)
     parser.add_argument("--host", default="localhost", type=str)

diff --git a/moshi_mlx/pyproject.toml b/moshi_mlx/pyproject.toml
@@ -6,7 +6,7 @@ dependencies = [
     "numpy >= 2.1.0, < 2.2",
     "safetensors >= 0.4.0, < 0.5",
     "huggingface-hub >= 0.24, < 0.25",
-    "rustymimi == 0.1.1",
+    "rustymimi == 0.2.2",
     "sentencepiece == 0.2",
     "sounddevice == 0.5",
     "sphn >= 0.1.4",

diff --git a/moshi_mlx/requirements.txt b/moshi_mlx/requirements.txt
@@ -9,7 +9,7 @@ numpy==2.1.0
 psutil==6.0.0
 pycparser==2.22
 pyright==1.1.378
-rustymimi==0.1.1
+rustymimi==0.2.2
 safetensors==0.4.4
 sentencepiece==0.2.0
 six==1.16.0