"docs/source/ko/using-diffusers/img2img.md" did not exist on "d2a5247a1f919e1dfbf2a15f2fda21a1ea11a116"
Unverified Commit c550e52f authored by 文峰's avatar 文峰 Committed by GitHub
Browse files

Fix scheduler proctitle suffix is ​​None (#4326)


Co-authored-by: default avatarwenfeng.wf <wenfeng.wf@alibaba-inc.com>
parent e35a93fa
......@@ -1677,9 +1677,16 @@ def run_scheduler_process(
dp_rank: Optional[int],
pipe_writer,
):
# Generate the prefix
if dp_rank is None:
prefix = f" TP{tp_rank}"
else:
prefix = f" DP{dp_rank} TP{tp_rank}"
# Config the process
# kill_itself_when_parent_died() # This is disabled because it does not work for `--dp 2`
setproctitle.setproctitle(f"sglang::scheduler_{dp_rank}")
setproctitle.setproctitle(f"sglang::scheduler{prefix.replace(' ', '_')}")
faulthandler.enable()
parent_process = psutil.Process().parent()
......@@ -1688,10 +1695,6 @@ def run_scheduler_process(
dp_rank = int(os.environ["SGLANG_DP_RANK"])
# Configure the logger
if dp_rank is None:
prefix = f" TP{tp_rank}"
else:
prefix = f" DP{dp_rank} TP{tp_rank}"
configure_logger(server_args, prefix=prefix)
suppress_other_loggers()
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment