mirror of
https://git.datalinker.icu/vllm-project/vllm.git
synced 2026-03-19 22:07:42 +08:00
updatedd
Signed-off-by: Robert Shaw <robshaw@redhat.com>
This commit is contained in:
parent
2a68433a82
commit
1ced153eec
@ -961,6 +961,8 @@ class DPAsyncMPClient(AsyncMPClient):
|
||||
counts, wave, running = msgspec.msgpack.decode(buf)
|
||||
self.current_wave = wave
|
||||
self.engines_running = running
|
||||
# NOTE: counts includes num running for all global
|
||||
# EngineCores, so need to slide for the local ones.
|
||||
self.lb_engines = counts[dp_start_rank:dp_end_rank]
|
||||
|
||||
resources.stats_update_task = asyncio.create_task(
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user