vllm

dd2a94fd - fix assert error num_tokens_across_dp is None

fix assert error num_tokens_across_dp is None or num_tokens_across_dp[dp_rank] == batchsize Signed-off-by: yewentao256 <zhyanwentao@126.com>