from trapper.models.auto_wrappers import (
    ModelWrapperForCausalLM,
    ModelWrapperForMaskedLM,
    ModelWrapperForMultipleChoice,
    ModelWrapperForNextSentencePrediction,
    ModelWrapperForQuestionAnswering,
    ModelWrapperForSeq2SeqLM,
    ModelWrapperForSequenceClassification,
    ModelWrapperForTokenClassification,
)
from trapper.models.model_wrapper import ModelWrapper
