mindformers.models
===================

.. automodule:: mindformers.models

mindformers.models
-------------------

.. autosummary::
    :toctree: models
    :nosignatures:
    :template: classtemplate.rst

    mindformers.models.BaseConfig
    mindformers.models.BaseImageProcessor
    mindformers.models.BaseModel
    mindformers.models.BaseProcessor
    mindformers.models.Tokenizer

mindformers.models.bert
------------------------

.. autosummary::
    :toctree: models
    :nosignatures:
    :template: classtemplate.rst

    mindformers.models.bert.BertConfig
    mindformers.models.bert.BertModel
    mindformers.models.bert.BertForMultipleChoice
    mindformers.models.bert.BertForPreTraining
    mindformers.models.bert.BertForQuestionAnswering
    mindformers.models.bert.BertForTokenClassificatio
    mindformers.models.bert.BertTokenizer
    mindformers.models.bert.BertProcessor

mindformers.models.t5
------------------------

.. autosummary::
    :toctree: models
    :nosignatures:
    :template: classtemplate.rst

    mindformers.models.t5.T5Config
    mindformers.models.t5.T5ForConditionalGeneration
    mindformers.models.t5.T5Processor
    mindformers.models.t5.T5Tokenizer

mindformers.models.clip
------------------------

.. autosummary::
    :toctree: models
    :nosignatures:
    :template: classtemplate.rst

    mindformers.models.clip.CLIPConfig
    mindformers.models.clip.CLIPVisionConfig
    mindformers.models.clip.CLIPTextConfig
    mindformers.models.clip.CLIPModel
    mindformers.models.clip.CLIPTokenizer
    mindformers.models.clip.CLIPProcessor
    mindformers.models.clip.CLIPImageProcessor

mindformers.models.mae
------------------------

.. autosummary::
    :toctree: models
    :nosignatures:
    :template: classtemplate.rst

    mindformers.models.mae.ViTMAEConfig
    mindformers.models.mae.ViTMAEForPreTraining
    mindformers.models.mae.ViTMAEProcessor
    mindformers.models.mae.ViTMAEImageProcessor

mindformers.models.swin
------------------------

.. autosummary::
    :toctree: models
    :nosignatures:
    :template: classtemplate.rst

    mindformers.models.swin.SwinConfig
    mindformers.models.swin.SwinModel
    mindformers.models.swin.SwinForImageClassification
    mindformers.models.swin.SwinImageProcessor
    mindformers.models.swin.SwinProcessor

mindformers.models.vit
------------------------

.. autosummary::
    :toctree: models
    :nosignatures:
    :template: classtemplate.rst

    mindformers.models.vit.ViTConfig
    mindformers.models.vit.ViTModel
    mindformers.models.vit.ViTProcessor
    mindformers.models.vit.ViTImageProcessor
    mindformers.models.vit.ViTForImageClassification

mindformers.models.gpt2
------------------------

.. autosummary::
    :toctree: models
    :nosignatures:
    :template: classtemplate.rst

    mindformers.models.gpt2.GPT2Config
    mindformers.models.gpt2.GPT2Model
    mindformers.models.gpt2.GPT2LMHeadModel
    mindformers.models.gpt2.GPT2Tokenizer
    mindformers.models.gpt2.GPT2Processor

mindformers.models.glm
------------------------

.. autosummary::
    :toctree: models
    :nosignatures:
    :template: classtemplate.rst

    mindformers.models.glm.GLMConfig
    mindformers.models.glm.GLMChatModel
    mindformers.models.glm.GLMForPreTraining
    mindformers.models.glm.GLMForPreTrainingWithLora
    mindformers.models.glm.GLMChatModelWithLora
    mindformers.models.glm.ChatGLMTokenizer
    mindformers.models.glm.GLMProcessor

mindformers.models.llama
------------------------

.. autosummary::
    :toctree: models
    :nosignatures:
    :template: classtemplate.rst

    mindformers.models.llama.LlamaConfig
    mindformers.models.llama.LlamaModel
    mindformers.models.llama.LlamaForCausalLM
    mindformers.models.llama.LlamaForCausalLMWithLora
    mindformers.models.llama.LlamaTokenizer
    mindformers.models.llama.LlamaProcessor

mindformers.models.bloom
------------------------

.. autosummary::
    :toctree: models
    :nosignatures:
    :template: classtemplate.rst

    mindformers.models.bloom.BloomConfig
    mindformers.models.bloom.BloomModel
    mindformers.models.bloom.BloomLMHeadModel
    mindformers.models.bloom.BloomTokenizer
    mindformers.models.bloom.BloomProcessor