diff --git a/python/mlx/_distributed_utils/launch.py b/python/mlx/_distributed_utils/launch.py index f87f112d6..56c7323e9 100644 --- a/python/mlx/_distributed_utils/launch.py +++ b/python/mlx/_distributed_utils/launch.py @@ -107,6 +107,9 @@ class RemoteProcess(CommandProcess): def make_launch_script(rank, cwd, files, env, command): script = "" + # Disable echo + script = "stty -echo; " + # Write the PID to a file so we can kill the process if needed script += "pidfile=$(mktemp); " script += "echo $$ > $pidfile; "