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.ChatGLM2ForConditionalGeneration
mindformers.models.ChatGLM2Config
mindformers.models.ChatGLM3Tokenizer
mindformers.models.ChatGLM4Tokenizer