misc: updated mlx in multi-machine example

prefix-dev · Aug 8, 2024 · e6bea41 · e6bea41
1 parent 094ba23
commit e6bea41
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 12 deletions.
diff --git a/examples/multi-machine/pixi.lock b/examples/multi-machine/pixi.lock
diff --git a/examples/multi-machine/pixi.toml b/examples/multi-machine/pixi.toml
@@ -33,15 +33,15 @@ pytorch-cuda = { version = "12.1.*", channel = "pytorch" }
 
 [feature.mlx]
 platforms = ["osx-arm64"]
-system-requirements = { macos = "13.3" }
+system-requirements = { macos = "13.5" }
+
+[feature.mlx.dependencies]
+mlx = ">=0.16.1,<0.17"
 
 [feature.mlx.tasks]
 test = "python test.py --mlx"
 train = "python train.py --mlx"
 
-[feature.mlx.dependencies]
-mlx = "*"
-
 [environments]
 cuda = ["cuda"]
 mlx = ["mlx"]