mindformers

Auto Classes

mindformers.AutoConfig	This is a generic configuration class that will be instantiated as one of the configuration classes of the library when created with the from_pretrained() class method.
mindformers.AutoModel	This is a generic model class that will be instantiated as one of the base model classes of the library when created with the AutoModel.from_pretrained class method or the AutoModel.from_config class method.
mindformers.AutoModelForCausalLM	This is a generic model class that will be instantiated as one of the model classes of the library (with a causal language modeling head) when created with the AutoModelForCausalLM.from_pretrained class method or the AutoModelForCausalLM.from_config class method.
mindformers.AutoProcessor	This is a generic processor class that will be instantiated as one of the processor classes of the library when created with the from_pretrained() class method.
mindformers.AutoTokenizer	This is a generic tokenizer class that will be instantiated as one of the tokenizer classes of the library when created with the from_pretrained class method.

Trainer High-level Interface

mindformers.Trainer	Executor of general task trainers.
mindformers.TrainingArguments	TrainingArguments is the subset of the arguments we use in our default config which is related to the training in MindSpore.

pipeline High-level Interface

mindformers.pipeline

Pipeline for executing the inference flow of integrated tasks and models.

Run Check

mindformers.run_check

Check whether the installed CANN, driver, MindSpore and MindFormers versions are matched.

Service-oriented

mindformers.ModelRunner

ModelRunner API, supports MindFormers to be a backend of MindIEServer.