# Copyright (c) 2022 Agora
# Licensed under The MIT License [see LICENSE for details]
from zeta.architecture.decoder import Decoder
from zeta.architecture.config import DecoderConfig, EncoderConfig, EncoderDecoderConfig
from zeta.architecture.encoder_decoder import EncoderDecoder


#models
from zeta.models.KosmosX import KosmosTokenizer, Kosmos
from zeta.models.LongNet import LongNetTokenizer, LongNet


#attention
from zeta.utils.attention.dilated_attention import DilatedAttention2