Ensure metrics are logged regardless of requests (#2347)
This commit is contained in:
committed by
GitHub
parent
937e7b7d7c
commit
d0215a58e7
@@ -506,3 +506,9 @@ class AsyncLLMEngine:
|
||||
max_log_len=engine_args.max_log_len,
|
||||
start_engine_loop=start_engine_loop)
|
||||
return engine
|
||||
|
||||
async def do_log_stats(self) -> None:
|
||||
if self.engine_use_ray:
|
||||
await self.engine.do_log_stats.remote()
|
||||
else:
|
||||
self.engine.do_log_stats()
|
||||
|
||||
@@ -641,6 +641,9 @@ class LLMEngine:
|
||||
|
||||
return self._process_model_outputs(output, scheduler_outputs)
|
||||
|
||||
def do_log_stats(self) -> None:
|
||||
self._log_system_stats(False, 0)
|
||||
|
||||
def _log_system_stats(
|
||||
self,
|
||||
prompt_run: bool,
|
||||
|
||||
Reference in New Issue
Block a user