From c6286c9acf84f6c1a25cd3cffd641f123f524fa9 Mon Sep 17 00:00:00 2001 From: Hongsheng Liu Date: Mon, 24 Aug 2026 23:56:57 +0800 Subject: [PATCH] [CI] Resume AR admission after sleep-mode wake Signed-off-by: Hongsheng Liu --- tests/entrypoints/test_omni_sleep_mode.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/tests/entrypoints/test_omni_sleep_mode.py b/tests/entrypoints/test_omni_sleep_mode.py index e71a13b1f..10d614dea 100644 --- a/tests/entrypoints/test_omni_sleep_mode.py +++ b/tests/entrypoints/test_omni_sleep_mode.py @@ -371,6 +371,8 @@ class TestOmniDiffusionSleepMode: # --- wake + VRAM restore to initial (±3 GiB) --- logger.info("Waking up (Reloading Weights)...") await diffusion_engine.wake_up(stage_ids=[0]) + # AR stages keep the frontend admission gate held after wake. + await diffusion_engine.resume_generation(stage_ids=[0]) await asyncio.sleep(2.0) gc.collect() get_vram_info(device_id)