fix dp_rank env (#2144)

sgl-project · Nov 23, 2024 · 52f58fc · 52f58fc
1 parent 145c0dd
commit 52f58fc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/sglang/srt/managers/scheduler.py b/python/sglang/srt/managers/scheduler.py
@@ -1389,7 +1389,7 @@ def run_scheduler_process(
 ):
     # [For Router] if env var "DP_RANK" exist, set dp_rank to the value of the env var
     if dp_rank is None and "DP_RANK" in os.environ:
-        dp_rank = int(os["DP_RANK"])
+        dp_rank = int(os.environ["DP_RANK"])
 
     if dp_rank is None:
         configure_logger(server_args, prefix=f" TP{tp_rank}")