Skip to content

Commit fe2f845

Browse files
committed
add timeout for llama4
Signed-off-by: Xin He (SW-GPU) <[email protected]>
1 parent 989c25f commit fe2f845

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

tests/integration/defs/accuracy/test_llm_api_pytorch.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -644,6 +644,7 @@ def test_nvfp4_tp4(self):
644644
extra_evaluator_kwargs=dict(apply_chat_template=True))
645645

646646

647+
@pytest.mark.timeout(14400)
647648
class TestLlama4MaverickInstruct(LlmapiAccuracyTestHarness):
648649
MODEL_NAME = "meta-llama/Llama-4-Maverick-17B-128E-Instruct"
649650
MODEL_PATH = f"{llm_models_root()}/llama4-models/Llama-4-Maverick-17B-128E-Instruct"
@@ -1896,7 +1897,7 @@ def test_guided_decoding_4gpus(self, backend: str, mtp_nextn: int, mocker):
18961897
task.evaluate(llm)
18971898

18981899

1899-
@pytest.mark.timeout(7200)
1900+
@pytest.mark.timeout(14400)
19001901
@pytest.mark.skip_less_device_memory(80000)
19011902
class TestDeepSeekR1(LlmapiAccuracyTestHarness):
19021903
MODEL_NAME = "deepseek-ai/DeepSeek-R1"

0 commit comments

Comments
 (0)