Unverified Commit 52f58fc4 authored by Byron Hsu's avatar Byron Hsu Committed by GitHub
Browse files

fix dp_rank env (#2144)

parent 145c0ddc
......@@ -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}")
......
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