AI开发平台MODELARTS-SD1.5基于DevServer适配MindSpore-Lite NPU推理指导(6.3.T041):Step3 下载原始模型包
Step3 下载原始模型包
从HuggingFace官网下载SD1.5模型包到本地,下载地址:https://huggingface.co/runwayml/stable-diffusion-v1-5/tree/onnx。
下载如下图所示4个目录的模型,并将其放在插件的对应目录中。
模型包目录结构如下,将下载后的模型按照如下目录上传到对应文件夹中。
/home/ma-user/sdv1-5_ms_code/onnx/ #插件包解压后的目录 |- || safety_checker #目录safety_checker需要手动创建 |-- model.onnx 模型 |- || text_encoder #目录text_encoder需要手动创建 |-- model.onnx 模型 |- || unet |-- config.ini 配置文件 |-- model.onnx 模型 |-- weights.pb 模型 |- || vae_decoder |-- config.ini 配置文件 |-- model.onnx 模型 |- - ******
unet和vae_decoder目录不需要创建,text_encoder和safety_checker目录需要手动创建,命令如下。
cd /home/ma-user/sdv1-5_ms_code/onnx mkdir text_encoder safety_checker