[NPU] Support radix-cache with mamba-extra-buffer for Qwen3.5 (#23891)
This commit is contained in:
@@ -2465,8 +2465,8 @@ class ServerArgs:
|
|||||||
)
|
)
|
||||||
|
|
||||||
assert (
|
assert (
|
||||||
is_cuda() or is_musa()
|
is_cuda() or is_musa() or is_npu()
|
||||||
), "Mamba extra_buffer is only supported on CUDA and MUSA devices with FLA backend"
|
), "Mamba extra_buffer is only supported on CUDA and MUSA and NPU devices with FLA backend"
|
||||||
if self.speculative_num_draft_tokens is not None:
|
if self.speculative_num_draft_tokens is not None:
|
||||||
assert (
|
assert (
|
||||||
self.mamba_track_interval >= self.speculative_num_draft_tokens
|
self.mamba_track_interval >= self.speculative_num_draft_tokens
|
||||||
|
|||||||
Reference in New Issue
Block a user