diff --git a/test/registered/8-gpu-models/test_qwen3_235b.py b/test/registered/8-gpu-models/test_qwen3_235b.py index 25cc4173b..c0e5058c9 100644 --- a/test/registered/8-gpu-models/test_qwen3_235b.py +++ b/test/registered/8-gpu-models/test_qwen3_235b.py @@ -31,10 +31,10 @@ class TestQwen3235BFP8(unittest.TestCase): """Run performance and accuracy for Qwen3-235B-FP8.""" base_args = [ "--tp=8", + "--ep=2", "--trust-remote-code", ] eagle3_args = [ - "--ep=2", "--speculative-algorithm=EAGLE3", f"--speculative-draft-model-path={QWEN3_235B_EAGLE3_MODEL_PATH}", "--speculative-num-steps=3",