Add KV4-capable backend flashmla and update server args (#14989)
Signed-off-by: Ho-Ren (Jack) Chuang <horenchuang@bytedance.com>
This commit is contained in:
committed by
GitHub
parent
4b7b5af36a
commit
171b442ad3
@@ -1607,6 +1607,7 @@ class ServerArgs:
|
||||
"cutlass_mla",
|
||||
"flashinfer",
|
||||
"trtllm_mla",
|
||||
"flashmla",
|
||||
]
|
||||
assert (
|
||||
self.attention_backend in KV4_ATTENTION_MLA_BACKEND_CHOICES
|
||||
|
||||
Reference in New Issue
Block a user