Skip to content

GitLab

Unverified Commit 751c3a03 authored by Lianmin Zheng's avatar Lianmin Zheng Committed by GitHub
Browse files

Fix dp print message (#2138)

parent 60769be1
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
python/sglang/srt/managers/scheduler.py
View file @ 751c3a03
...@@ -1388,8 +1388,8 @@ def run_scheduler_process( ...@@ -1388,8 +1388,8 @@ def run_scheduler_process(
pipe_writer, pipe_writer,
): ):
# [For Router] if env var "DP_RANK" exist, set dp_rank to the value of the env var # [For Router] if env var "DP_RANK" exist, set dp_rank to the value of the env var
if dp_rank is None: if dp_rank is None and "DP_RANK" in os.environ:
dp_rank = int(os.getenv("DP_RANK", -1)) dp_rank = int(os["DP_RANK"])
if dp_rank is None: if dp_rank is None:
configure_logger(server_args, prefix=f" TP{tp_rank}") configure_logger(server_args, prefix=f" TP{tp_rank}")
......
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