assert self.batched_router_logits.size(-1) == full_router_logits.size(-1)
Signed-off-by: yewentao256 <zhyanwentao@126.com>
ruff
yapf
isort
rst
md
vllm.utils