1fee098f10
Signed-off-by: <> Co-authored-by: William Lin <SolitaryThinker@users.noreply.github.com> Co-authored-by: Will Lin <wlsaidhi@gmail.com> Co-authored-by: Zhou, Wei <wzhou322@gatech.edu> Co-authored-by: Kevin Lin <42618777+kevin314@users.noreply.github.com> Co-authored-by: kevin314 <kevin.lin.cs1@gmail.com> Co-authored-by: JerryZhou54 <69577934+JerryZhou54@users.noreply.github.com> Co-authored-by: Yongqi Chen <144848849+BrianChen1129@users.noreply.github.com> Co-authored-by: Peiyuan Zhang <m2deng@ucsd.edu>
18 lines
597 B
Python
18 lines
597 B
Python
# SPDX-License-Identifier: Apache-2.0
|
|
|
|
from fastvideo.v1.attention.backends.abstract import (AttentionBackend,
|
|
AttentionMetadata,
|
|
AttentionMetadataBuilder)
|
|
from fastvideo.v1.attention.layer import DistributedAttention, LocalAttention
|
|
from fastvideo.v1.attention.selector import get_attn_backend
|
|
|
|
__all__ = [
|
|
"DistributedAttention",
|
|
"LocalAttention",
|
|
"AttentionBackend",
|
|
"AttentionMetadata",
|
|
"AttentionMetadataBuilder",
|
|
# "AttentionState",
|
|
"get_attn_backend",
|
|
]
|