mindformers

Auto类

mindformers.AutoConfig

这是一个通用的配置类,当使用 from_pretrained() 类方法时,它会自动实例化模型Config类,并返回。

mindformers.AutoModel

这是一个通用的模型类,使用类方法 AutoModel.from_pretrainedAutoModel.from_config 创建时会自动实例化为库中的基础模型类之一。

mindformers.AutoModelForCausalLM

这是一个通用的模型类,使用类方法 AutoModelForCausalLM.from_pretrainedAutoModelForCausalLM.from_config 创建时会自动实例化为库中(带有因果语言建模头)的模型类之一。

mindformers.AutoModelForZeroShotImageClassification

这是一个通用的模型类,使用类方法 AutoModelForZeroShotImageClassification.from_pretrainedAutoModelForZeroShotImageClassification.from_config 创建时会自动实例化为库中(带有零样本图片分类建模头)的模型类之一。

mindformers.AutoProcessor

这是一个通用的Processor类,当使用 from_pretrained() 类方法时,它会自动实例化Processor类,并返回。

mindformers.AutoTokenizer

这是一个通用的Tokenizer类,当使用 from_pretrained() 类方法时,它会自动实例化Tokenzier类,并返回。

Trainer高阶接口

mindformers.Trainer

Trainer是通用的任务执行组件,通过参数中的task名称或者配置文件可以指定任务并且初始化一个与该任务相关的trainer实例。

mindformers.TrainingArguments

TrainingArguments是与MindSpore训练相关的参数集合。

pipeline高阶接口

mindformers.pipeline

通过流水线执行套件中已集成任务和模型的推理流程。

运行检查

mindformers.run_check

检查CANN、固件与驱动、MindSpore和MindFormers安装版本是否匹配。

面向服务化

mindformers.ModelRunner

用于将 MindFormers 的一个模型运行实例作为 MindIEServer 的后端。