"""Module implementing MPT architecture for downstream task GPT1-style."""

from mt_dao.models.mpt.mpt_for_question_answering import MPTForQuestionAnswering
from mt_dao.models.mpt.mpt_for_sequence_classification import (
    MPTForSequenceClassification,
)
from mt_dao.models.mpt.mpt_for_token_classification import MPTForTokenClassification
from mt_dao.models.mpt.mpt_mup import ComposerMPTMuPCausalLM, MPTMuPForCausalLM

__all__ = [
    "ComposerMPTMuPCausalLM",
    "MPTForQuestionAnswering",
    "MPTForSequenceClassification",
    "MPTForTokenClassification",
    "MPTMuPForCausalLM",
]
