From d84aa6246a5f6eba2fd7b72c10a87d8f4503eca9 Mon Sep 17 00:00:00 2001 From: sibianl Date: Fri, 10 Oct 2025 20:03:26 +0800 Subject: [PATCH] fix(node): change ascii animation display time --- src/parallax/launch.py | 7 ++++++- src/parallax/server/executor.py | 2 -- 2 files changed, 6 insertions(+), 3 deletions(-) diff --git a/src/parallax/launch.py b/src/parallax/launch.py index 64bc9552..1d2743e3 100644 --- a/src/parallax/launch.py +++ b/src/parallax/launch.py @@ -23,6 +23,7 @@ from parallax.server.http_server import launch_http_server from parallax.server.server_args import parse_args from parallax.utils.utils import get_current_device +from parallax_utils.ascii_anime import display_parallax_join from parallax_utils.logging_config import get_logger logger = get_logger("parallax.launch") @@ -47,7 +48,6 @@ try: args = parse_args() logger.debug(f"args: {args}") - args.recv_from_peer_addr = f"ipc://{tempfile.NamedTemporaryFile().name}" args.send_to_peer_addr = f"ipc://{tempfile.NamedTemporaryFile().name}" args.executor_input_ipc = f"ipc://{tempfile.NamedTemporaryFile().name}" @@ -62,6 +62,8 @@ args.model_path = mlx_model_repo logger.debug(f"Replace mlx model path: {mlx_model_repo}") if args.scheduler_addr is None: + display_parallax_join(args.model_path) + # only launch http server on head node if args.start_layer == 0: http_server_process = launch_http_server(args) @@ -118,6 +120,9 @@ f"Start Executor with start_layer: {args.start_layer}, end_layer: {args.end_layer}" ) gradient_server.status = ServerState.INITIALIZING + + display_parallax_join(args.model_path) + # only launch http server on head node if args.start_layer == 0: http_server_process = launch_http_server(args) diff --git a/src/parallax/server/executor.py b/src/parallax/server/executor.py index b0900371..f449f582 100644 --- a/src/parallax/server/executor.py +++ b/src/parallax/server/executor.py @@ -56,7 +56,6 @@ pad_inputs, pad_prefix_caches, ) -from parallax_utils.ascii_anime import display_parallax_join from parallax_utils.logging_config import get_logger from parallax_utils.utils import compute_max_batch_size @@ -274,7 +273,6 @@ def __init__( self.send_to_ipc_socket = get_zmq_socket( self.zmq_context, zmq.PUSH, executor_output_ipc_addr, bind=False ) - display_parallax_join(model_repo) @classmethod def create_from_args(cls, args: argparse.Namespace):