Skip to content

GitLab

"docs/source/using-diffusers/rl.mdx" did not exist on "a0520193e15951655ee2c08c24bfdca716f6f64c"
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
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",
......
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