from speculators.models.attention import ALL_ATTENTION_FUNCTIONS  # noqa: F401

from .dflash import DFlashDraftModel, DFlashSpeculatorConfig
from .dflash2 import DFlash2DraftModel, DFlash2SpeculatorConfig
from .dfly import DFlyDraftModel, DFlySpeculatorConfig
from .domino import DominoDraftModel, DominoSpeculatorConfig
from .dspark import DSparkDraftModel, DSparkSpeculatorConfig
from .eagle3 import Eagle3DraftModel, Eagle3SpeculatorConfig
from .mtp import MTPDraftModel, MTPSpeculatorConfig
from .ngram_candidates import NGramCandidatesDraftModel, NGramCandidatesSpeculatorConfig
from .ngram_dflash import NGramDFlashDraftModel, NGramDFlashSpeculatorConfig
from .peagle import PEagleDraftModel, PEagleSpeculatorConfig

__all__ = [
    "DFlash2DraftModel",
    "DFlash2SpeculatorConfig",
    "DFlashDraftModel",
    "DFlashSpeculatorConfig",
    "DFlyDraftModel",
    "DFlySpeculatorConfig",
    "DSparkDraftModel",
    "DSparkSpeculatorConfig",
    "DominoDraftModel",
    "DominoSpeculatorConfig",
    "Eagle3DraftModel",
    "Eagle3SpeculatorConfig",
    "MTPDraftModel",
    "MTPSpeculatorConfig",
    "NGramCandidatesDraftModel",
    "NGramCandidatesSpeculatorConfig",
    "NGramDFlashDraftModel",
    "NGramDFlashSpeculatorConfig",
    "PEagleDraftModel",
    "PEagleSpeculatorConfig",
]