昇腾社区首页
中文
注册

类说明

OpenSora推理解码模型,继承自class mindiesd.DiffusionModel,该函数在MindIE SD仓的路径为:mindiesd/models/vae.py。

成员

成员名称

描述

__init__

类初始化函数。

decode

视频帧解码函数。

get_latent_size

获取latent size。

config_class

初始化使用的config类,默认值为VideoAutoencoderConfig。

weights_name

对应加载权重文件的名称,默认值为'model.safetensors'。