Spaces:
Running
on
Zero
Running
on
Zero
File size: 1,325 Bytes
c8b42eb |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 |
from uniception.models.info_sharing.alternating_attention_transformer import (
MultiViewAlternatingAttentionTransformer,
MultiViewAlternatingAttentionTransformerIFR,
)
from uniception.models.info_sharing.cross_attention_transformer import (
MultiViewCrossAttentionTransformer,
MultiViewCrossAttentionTransformerIFR,
MultiViewTransformerInput,
)
from uniception.models.info_sharing.diff_cross_attention_transformer import (
DifferentialMultiViewCrossAttentionTransformer,
DifferentialMultiViewCrossAttentionTransformerIFR,
)
from uniception.models.info_sharing.global_attention_transformer import (
MultiViewGlobalAttentionTransformer,
MultiViewGlobalAttentionTransformerIFR,
)
INFO_SHARING_CLASSES = {
"cross_attention": (MultiViewCrossAttentionTransformer, MultiViewCrossAttentionTransformerIFR),
"diff_cross_attention": (
DifferentialMultiViewCrossAttentionTransformer,
DifferentialMultiViewCrossAttentionTransformerIFR,
),
"alternating_attention": (
MultiViewAlternatingAttentionTransformer,
MultiViewAlternatingAttentionTransformerIFR,
),
"global_attention": (
MultiViewGlobalAttentionTransformer,
MultiViewGlobalAttentionTransformerIFR,
),
}
__all__ = ["INFO_SHARING_CLASSES", "MultiViewTransformerInput"]
|