Merge pull request #49 from argmaxinc/arda/pin_mlx_0.17.3

Pin mlx==0.17.3

Author: arda-argmax

setup.py

@@ -3,7 +3,7 @@
 from setuptools import find_packages, setup
 from setuptools.command.install import install
 
-VERSION = "0.5.1"
+VERSION = "0.5.2"
 
 
 class VersionInstallCommand(install):
@@ -29,7 +29,7 @@ def run(self):
         "argmaxtools>=0.1.13",
         "torch",
         "safetensors",
-        "mlx>=0.17.1",
+        "mlx==0.17.3",
         "jaxtyping",
         "transformers",
         "pillow",