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