昇腾社区首页
中文
注册

类说明

MindIE SD模型基类,提供from_pretrained自动加载config和权重功能,该函数在MindIE SD仓的路径为:mindiesd/models/model_utils.py。

成员

成员名称

描述

config_class

model对应的config参数类。

weights_name

model权重的名称。

__init__

初始化函数。

dtype

参数的数据类型。

device

进行具体计算任务的硬件设备。

from_pretrained

模型自动初始化加载函数。