昇腾社区首页
中文
注册

类说明

OpenSora推理DiT模型参数配置,继承自DiffusionModel,该函数在MindIE SD仓的路径为:mindiesd/models/stdit3.py。

成员

成员名称

描述

config_class

加载config的类对象,默认为STDiT3Config。

weights_name

权重文件名称,默认为'model.safetensors'。

__init__

类初始化函数。

forward

生成视频推理函数。