[CI] Resume AR admission after sleep-mode wake

Signed-off-by: Hongsheng Liu <liuhongsheng4@huawei.com>
This commit is contained in:
Hongsheng Liu committed 2026-08-24 23:56:57 +08:00
1 parent 678aa25864
commit c6286c9acf
1 file changed
+2
@@ -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)