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
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

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.