[Fix] Add speculative_draft_model_revision to server_args (#5255)

Signed-off-by: Devashish Lal <devashish@rivosinc.com>
2025-09-05 04:45:46 -07:00
parent df97b31f37
commit 13705dae06
13 changed files with 68 additions and 45 deletions
--- a/test/srt/test_hybrid_attn_backend.py
+++ b/test/srt/test_hybrid_attn_backend.py
@@ -121,7 +121,7 @@ class TestHybridAttnBackendSpeculativeDecoding(TestHybridAttnBackendBase):
        return DEFAULT_SERVER_ARGS + [
            "--speculative-algorithm",
            "EAGLE",
-            "--speculative-draft",
+            "--speculative-draft-model-path",
            DEFAULT_EAGLE_DRAFT_MODEL_FOR_TEST,
            "--speculative-num-steps",
            "3",