昇腾社区首页
中文
注册

def __init__

函数功能

类初始化函数。

函数原型

def __init__(self, from_pretrained, set_patch_parallel=False, **kwargs):

参数说明

参数名

输入/输出

说明

from_pretrained

输入

加载vae_2d模型的权重路径。

set_patch_parallel

输入

是否开启Conv3D并行优化。

默认为False。

kwargs

输入

额外参数。

返回值说明