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

Base Classes
--------------

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

    mindformers.models.PreTrainedModel
    mindformers.models.PretrainedConfig
    mindformers.models.PreTrainedTokenizer
    mindformers.models.PreTrainedTokenizerFast
    mindformers.models.multi_modal.ModalContentTransformTemplate

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

    mindformers.models.LlamaForCausalLM
    mindformers.models.LlamaConfig
    mindformers.models.LlamaTokenizer
    mindformers.models.LlamaTokenizerFast

GLM
---------------------

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

    mindformers.models.glm2.ChatGLM2ForConditionalGeneration
    mindformers.models.glm2.ChatGLM2Config
    mindformers.models.glm2.ChatGLM3Tokenizer
    mindformers.models.glm2.ChatGLM4Tokenizer