From 2c2b19b18ba7ee24e321d692ba84a69fa6cf532d Mon Sep 17 00:00:00 2001 From: Liangsheng Yin Date: Mon, 8 Sep 2025 18:16:52 +0800 Subject: [PATCH] [CI] fix ambiguous argument in testing hybrid attentions. (#10161) --- test/srt/test_hybrid_attn_backend.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/test/srt/test_hybrid_attn_backend.py b/test/srt/test_hybrid_attn_backend.py index 306259df9..cd93f434d 100644 --- a/test/srt/test_hybrid_attn_backend.py +++ b/test/srt/test_hybrid_attn_backend.py @@ -142,7 +142,7 @@ class TestHybridAttnBackendSpeculativeDecodingPrefillBackend(TestHybridAttnBacke return DEFAULT_SERVER_ARGS + [ "--speculative-algorithm", "EAGLE", - "--speculative-draft", + "--speculative-draft-model-path", DEFAULT_EAGLE_DRAFT_MODEL_FOR_TEST, "--speculative-num-steps", "3", @@ -165,7 +165,7 @@ class TestHybridAttnBackendSpeculativeDecodingDecodeBackend(TestHybridAttnBacken return DEFAULT_SERVER_ARGS + [ "--speculative-algorithm", "EAGLE", - "--speculative-draft", + "--speculative-draft-model-path", DEFAULT_EAGLE_DRAFT_MODEL_FOR_TEST, "--speculative-num-steps", "3",