[CI] Resume AR admission after sleep-mode wake
Signed-off-by: Hongsheng Liu <liuhongsheng4@huawei.com>
This commit is contained in:
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)
|
||||
|
||||
Reference in new issue
Block a user