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。