Skip to content

GitLab

Unverified Commit 2c2b19b1 authored by Liangsheng Yin's avatar Liangsheng Yin Committed by GitHub
Browse files

[CI] fix ambiguous argument in testing hybrid attentions. (#10161)

parent 72f9fc5f
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
test/srt/test_hybrid_attn_backend.py
View file @ 2c2b19b1
...@@ -142,7 +142,7 @@ class TestHybridAttnBackendSpeculativeDecodingPrefillBackend(TestHybridAttnBacke ...@@ -142,7 +142,7 @@ class TestHybridAttnBackendSpeculativeDecodingPrefillBackend(TestHybridAttnBacke
return DEFAULT_SERVER_ARGS + [ return DEFAULT_SERVER_ARGS + [
"--speculative-algorithm", "--speculative-algorithm",
"EAGLE", "EAGLE",
"--speculative-draft", "--speculative-draft-model-path",
DEFAULT_EAGLE_DRAFT_MODEL_FOR_TEST, DEFAULT_EAGLE_DRAFT_MODEL_FOR_TEST,
"--speculative-num-steps", "--speculative-num-steps",
"3", "3",
...@@ -165,7 +165,7 @@ class TestHybridAttnBackendSpeculativeDecodingDecodeBackend(TestHybridAttnBacken ...@@ -165,7 +165,7 @@ class TestHybridAttnBackendSpeculativeDecodingDecodeBackend(TestHybridAttnBacken
return DEFAULT_SERVER_ARGS + [ return DEFAULT_SERVER_ARGS + [
"--speculative-algorithm", "--speculative-algorithm",
"EAGLE", "EAGLE",
"--speculative-draft", "--speculative-draft-model-path",
DEFAULT_EAGLE_DRAFT_MODEL_FOR_TEST, DEFAULT_EAGLE_DRAFT_MODEL_FOR_TEST,
"--speculative-num-steps", "--speculative-num-steps",
"3", "3",
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment