昇腾社区首页
中文
注册
开发者
下载

def from_pretrained

函数功能

自动从给定路径中加载pipeline。

model_path中的config文件,可以将scheduler、text_encoder、tokenizer、transformer、vae的模型配置为mindiesd、transformers、opensora等库的任意类。此时MindIE SD会调用第三方库的from_pretrained、from_config等函数,用户使用时需要注意相关安全风险,使用安全的输入参数配置,例如:trust_remote_code=False。

函数原型

1
def from_pretrained(cls, model_path, **kwargs):

参数说明

参数名

输入/输出

类型

说明

model_path

输入

str

模型路径。

该文件夹、子文件夹以及模型相关文件均会依据文件目录权限说明进行权限安全校验。

kwargs

输入

-

可变额外入参,如果kwargs中有类初始化需要的参数,则优先会使用kwargs中的入参。

返回值说明

初始化后的pipeline。