Skip to content

Commit 2f2a04b

Browse files
committed
readability \diffs
Signed-off-by: Patryk Saffer <[email protected]>
1 parent 3ca1e6c commit 2f2a04b

File tree

2 files changed

+3
-1
lines changed

2 files changed

+3
-1
lines changed

vllm/config/parallel.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -682,4 +682,5 @@ def _verify_args(self) -> Self:
682682
raise ValueError("save_load_window cannot be set to true with static eplb.")
683683
if self.eplb_config.use_async and self.eplb_config.static:
684684
raise ValueError("use_async cannot be set to true with static eplb.")
685+
685686
return self

vllm/distributed/eplb/eplb_state.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -251,6 +251,7 @@ def __init__(self, parallel_config: ParallelConfig, device: torch.device):
251251
self.model_states: dict[str, EplbModelState] = {}
252252
"""
253253
Current step in the sliding window.
254+
254255
Different from `expert_rearrangement_step`,
255256
each EP rank may have its own `expert_load_window_step`.
256257
"""
@@ -621,7 +622,7 @@ def step(
621622

622623
if ep_group.rank() == 0:
623624
logger.info(
624-
"EPLBS step: %d for model %s: avg_tokens=%.2f, "
625+
"EPLB step: %d for model %s: avg_tokens=%.2f, "
625626
"max_tokens=%d, balancedness=%.4f",
626627
self.expert_rearrangement_step,
627628
eplb_model_state.model_name,

0 commit comments

Comments
 (0)