Skip to content

Commit 1de0baa

Browse files
committed
fix
1 parent b5a73a0 commit 1de0baa

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

lightllm/server/api_cli.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -43,11 +43,11 @@ def make_argument_parser() -> argparse.ArgumentParser:
4343
help="p d mode, decode node used for kv move manager rpyc server port",
4444
)
4545
parser.add_argument(
46-
"--select_p_d_node_func",
46+
"--select_p_d_node_strategy",
4747
type=str,
4848
default="round_robin",
4949
choices=["random", "round_robin", "adaptive_load"],
50-
help="select p d node func, can be round_robin, random or adaptive_load",
50+
help="pd master use this strategy to select p d node, can be round_robin, random or adaptive_load",
5151
)
5252
parser.add_argument(
5353
"--config_server_host",

lightllm/server/httpserver_for_pd_master/manager.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -403,7 +403,7 @@ def __init__(self, args):
403403
self.prefill_nodes: List[PD_Client_Obj] = []
404404
self.decode_nodes: List[PD_Client_Obj] = []
405405
self.url_to_pd_nodes: Dict[str, PD_Client_Obj] = {}
406-
self.selector = create_selector(args.select_p_d_node_func, self)
406+
self.selector = create_selector(args.select_p_d_node_strategy, self)
407407
return
408408

409409
def register_pd(self, pd_info_json, websocket):

0 commit comments

Comments
 (0)