Refactor distributed config

This commit is contained in:
Angelos Katharopoulos
2025-12-09 05:58:44 -08:00
parent cd4b12ce1b
commit 405d30b6e5
4 changed files with 556 additions and 9 deletions

View File

@@ -266,7 +266,7 @@ if __name__ == "__main__":
entry_points = {
"console_scripts": [
"mlx.launch = mlx._distributed_utils.launch:main",
# "mlx.distributed_config = mlx.distributed_run:distributed_config",
"mlx.distributed_config = mlx._distributed_utils.config:main",
]
}
install_requires = []