17 lines
540 B
Python
17 lines
540 B
Python
# SPDX-License-Identifier: Apache-2.0
|
|
|
|
from fastvideo.attention.backends.abstract import (AttentionBackend, AttentionMetadata, AttentionMetadataBuilder)
|
|
from fastvideo.attention.layer import (DistributedAttention, DistributedAttention_VSA, LocalAttention)
|
|
from fastvideo.attention.selector import get_attn_backend
|
|
|
|
__all__ = [
|
|
"DistributedAttention",
|
|
"LocalAttention",
|
|
"DistributedAttention_VSA",
|
|
"AttentionBackend",
|
|
"AttentionMetadata",
|
|
"AttentionMetadataBuilder",
|
|
# "AttentionState",
|
|
"get_attn_backend",
|
|
]
|