biondizzle/vllm
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

[Bugfix] Fix EPLB state logging error (#31455)

Browse Source 
Signed-off-by: Tyler Michael Smith <tlrmchlsmth@gmail.com>
This commit is contained in:
Tyler Michael Smith
2026-01-04 23:06:28 -05:00
committed by GitHub
parent 98b8b3abaa
commit fe8a9fbd2e
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
vllm/distributed/eplb/eplb_state.py
View File

@@ -423,7 +423,7 @@ class EplbState:
# Set the policy based on the selected eplb algorithm type. # Set the policy based on the selected eplb algorithm type.
policy_type = self.parallel_config.eplb_config.policy policy_type = self.parallel_config.eplb_config.policy
self.policy = EPLB_POLICIES[policy_type] self.policy = EPLB_POLICIES[policy_type]
logger.debug("Selected EPLB policy: %d", policy_type) logger.debug("Selected EPLB policy: %s", policy_type)
if global_expert_load is not None: if global_expert_load is not None:
ep_group = get_ep_group().device_group ep_group = get_ep_group().device_group
assert global_expert_load.shape == ( assert global_expert_load.shape == (