chores: adjust the attn register param order (#30688)

Signed-off-by: Hank <hcc.mayday@gmail.com>
This commit is contained in:
Hank_
2025-12-17 19:58:16 +08:00
committed by GitHub
parent fb980eb2fd
commit 6482e3895b

View File

@@ -201,8 +201,8 @@ _MAMBA_ATTN_OVERRIDES: dict[MambaAttentionBackendEnum, str] = {}
def register_backend(
backend: AttentionBackendEnum | MambaAttentionBackendEnum,
is_mamba: bool = False,
class_path: str | None = None,
is_mamba: bool = False,
) -> Callable[[type], type]:
"""Register or override a backend implementation.