Skip to content

Commit 66c375c

Browse files
committed
don't double eplb prefix
Signed-off-by: Patryk Saffer <[email protected]>
1 parent abbeb70 commit 66c375c

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

vllm/distributed/eplb/eplb_state.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -539,12 +539,12 @@ def rearrange(
539539

540540
if (
541541
is_main_rank
542-
and self.eplb_save_dir is not None
542+
and self.save_dir is not None
543543
and not is_profile
544544
and self.expert_rearrangement_step > 0
545545
):
546546
save_eplb_state(
547-
global_expert_load_window, self.eplb_save_dir, self.save_dir_iter
547+
global_expert_load_window, self.save_dir, self.save_dir_iter
548548
)
549549
self.save_dir_iter += 1
550550

0 commit comments

Comments
 (0)