From 4e4458e7bf8b16f34b9f76e07af04c77c5889e2a Mon Sep 17 00:00:00 2001 From: Yueqian Lin <70319226+linyueqian@users.noreply.github.com> Date: Fri, 22 May 2026 19:46:14 -0400 Subject: [PATCH] [Bugfix][CI] Run Whisper validation on CPU for single-GPU runners (#3822) Signed-off-by: Yueqian Lin --- tests/helpers/media.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/tests/helpers/media.py b/tests/helpers/media.py index daf3a2ba4..5c76a9e1e 100644 --- a/tests/helpers/media.py +++ b/tests/helpers/media.py @@ -623,9 +623,11 @@ def _whisper_transcribe_in_current_process(output_path: str) -> str: if current_omni_platform.is_available(): n = current_omni_platform.get_device_count() - if n == 1: - device_index = 0 - elif n > 1: + # Single-GPU runners (e.g. the L4 nightly): the model server already + # occupies device 0. Loading Whisper there, once per concurrent + # request, competes for VRAM and OOMs. Only borrow an accelerator + # when a spare device exists; otherwise validate on CPU. + if n > 1: device_index = n - 1 if device_index is not None: