pin mlx==0.17.3

arda-argmax · arda-argmax · commit 7448b5a05c86 · 2024-12-06T13:56:28.000-08:00
diff --git a/setup.py b/setup.py
@@ -29,7 +29,7 @@ def run(self):
         "argmaxtools>=0.1.13",
         "torch",
         "safetensors",
-        "mlx>=0.17.1",
+        "mlx==0.17.3",
         "jaxtyping",
         "transformers",
         "pillow",
