From 6940d7ce9147c7fa5b4dc36444942d5eaa7e1f65 Mon Sep 17 00:00:00 2001
From: "Kevin H. Luu" <khluu000@gmail.com>
Date: Mon, 8 Dec 2025 15:34:04 -0800
Subject: [PATCH] push

Signed-off-by: Kevin H. Luu <khluu000@gmail.com>
---
 .buildkite/jobs/tests.yaml | 1 +
 1 file changed, 1 insertion(+)

diff --git a/.buildkite/jobs/tests.yaml b/.buildkite/jobs/tests.yaml
index 8000c5614..b70c132d7 100644
--- a/.buildkite/jobs/tests.yaml
+++ b/.buildkite/jobs/tests.yaml
@@ -23,6 +23,7 @@ steps:
 - label: "Omni Model Test with H100"
   timeout_in_minutes: 20
   gpu: h100
+  num_gpus: 2
   commands:
     - export VLLM_WORKER_MULTIPROC_METHOD=spawn
     - pytest -s -v tests/multi_stages_h100/
\ No newline at end of file