EngineX-Hygon/sglang
5
0
Fork 0
Code Issues Pull Requests Actions 7 Projects Releases Wiki Activity

Fix dp print message (#2138)

Browse Source
This commit is contained in:
Lianmin Zheng
2024-11-23 01:22:26 -08:00
committed by GitHub
parent 60769be14d
commit 751c3a037c
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
python/sglang/srt/managers/scheduler.py
View File

@@ -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}")