Fix dp print message (#2138)

751c3a03 · Lianmin Zheng · GitHub · 60769be1 · 751c3a03
Unverified Commit 751c3a03 authored Nov 23, 2024 by Lianmin Zheng Committed by GitHub Nov 23, 2024
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

python/sglang/srt/managers/scheduler.py python/sglang/srt/managers/scheduler.py +2 -2

No files found.
--- a/python/sglang/srt/managers/scheduler.py
+++ b/python/sglang/srt/managers/scheduler.py
@@ -1388,8 +1388,8 @@ def run_scheduler_process(
    pipe_writer,
 ):
    # [For Router] if env var "DP_RANK" exist, set dp_rank to the value of the env var
-    if dp_rank is None:
-        dp_rank = int(os.getenv("DP_RANK", -1))
+    if dp_rank is None and "DP_RANK" in os.environ:
+        dp_rank = int(os["DP_RANK"])

    if dp_rank is None:
        configure_logger(server_args, prefix=f" TP{tp_rank}")