Unverified Commit b922c2eb authored by Cyrus Leung's avatar Cyrus Leung Committed by GitHub
Browse files

[Bugfix] Fix entrypoints metrics tests (#18063)


Signed-off-by: default avatarDarkLight1337 <tlleungac@connect.ust.hk>
parent 00b14e0f
...@@ -150,10 +150,6 @@ async def build_async_engine_client( ...@@ -150,10 +150,6 @@ async def build_async_engine_client(
async with build_async_engine_client_from_engine_args( async with build_async_engine_client_from_engine_args(
engine_args, args.disable_frontend_multiprocessing) as engine: engine_args, args.disable_frontend_multiprocessing) as engine:
# Don't keep the dummy data in memory
await engine.reset_mm_cache()
yield engine yield engine
...@@ -189,6 +185,10 @@ async def build_async_engine_client_from_engine_args( ...@@ -189,6 +185,10 @@ async def build_async_engine_client_from_engine_args(
usage_context=usage_context, usage_context=usage_context,
disable_log_requests=engine_args.disable_log_requests, disable_log_requests=engine_args.disable_log_requests,
disable_log_stats=engine_args.disable_log_stats) disable_log_stats=engine_args.disable_log_stats)
# Don't keep the dummy data in memory
await async_llm.reset_mm_cache()
yield async_llm yield async_llm
finally: finally:
if async_llm: if async_llm:
......
...@@ -289,7 +289,7 @@ class EngineCore: ...@@ -289,7 +289,7 @@ class EngineCore:
def reset_mm_cache(self): def reset_mm_cache(self):
# NOTE: Since this is mainly for debugging, we don't attempt to # NOTE: Since this is mainly for debugging, we don't attempt to
# re-sync the internal caches (P0 processor, P0 mirror, P1 mirror) # re-sync the internal caches (P0 processor, P0 mirror, P1 mirror)
if self.scheduler.get_num_unfinished_requests(): if self.scheduler.has_unfinished_requests():
logger.warning("Resetting the multi-modal cache when requests are " logger.warning("Resetting the multi-modal cache when requests are "
"in progress may lead to desynced internal caches.") "in progress may lead to desynced internal caches.")
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment