diff --git a/src/exo/worker/engines/mlx/utils_mlx.py b/src/exo/worker/engines/mlx/utils_mlx.py index d4639694f..30cdcaf93 100644 --- a/src/exo/worker/engines/mlx/utils_mlx.py +++ b/src/exo/worker/engines/mlx/utils_mlx.py @@ -815,9 +815,7 @@ def mlx_force_oom2(bytes_alloc: int = 1024**5): # the default is 1 petabyte lol # allocate a lot z = mx.zeros(shape=(size, size), dtype=mx.float32) - t1 = [ - mx.random.uniform(shape=(size, size), dtype=mx.float32) for _ in range(10) - ] + t1 = [mx.random.uniform(shape=(size, size), dtype=mx.float32) for _ in range(2)] print("t1-run") mx.eval(*t1) print("t1-eval")