diff --git a/tests/entrypoints/openai/correctness/test_lmeval.py b/tests/entrypoints/openai/correctness/test_lmeval.py index 5b23b42390..aad1b5e062 100644 --- a/tests/entrypoints/openai/correctness/test_lmeval.py +++ b/tests/entrypoints/openai/correctness/test_lmeval.py @@ -71,8 +71,9 @@ def test_lm_eval_accuracy_v1_engine(): more_args = [] - # Limit compilation time for V1 - if current_platform.is_tpu(): + # Limit compilation time for V1 on TPU + # Avoid OOM on XPU + if current_platform.is_tpu() or current_platform.is_xpu(): more_args = ["--max-num-seqs", "64"] run_test(more_args)