mindformers
Auto Classes
This is a generic configuration class that will be instantiated as one of the configuration classes of the library when created with the |
|
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. |
|
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. |
|
This is a generic model class that will be instantiated as one of the model classes of the library (with a zero-shot image classification head) when created with the AutoModelForZeroShotImageClassification.from_pretrained class method or the AutoModelForZeroShotImageClassification.from_config class method. |
|
This is a generic processor class that will be instantiated as one of the processor classes of the library when created with the |
|
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
Executor of general task trainers. |
|
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
Pipeline for executing the inference flow of integrated tasks and models. |
Run Check
Check whether the installed CANN, driver, MindSpore and MindFormers versions are matched. |
Service-oriented
ModelRunner API, supports MindFormers to be a backend of MindIEServer. |