AI开发平台MODELARTS-SD1.5基于DevServer适配MindSpore-Lite NPU推理指导(6.3.T041):Step3 下载原始模型包

时间:2024-07-06 17:52:03

Step3 下载原始模型包

从HuggingFace官网下载SD1.5模型包到本地,下载地址:https://huggingface.co/runwayml/stable-diffusion-v1-5/tree/onnx

下载如下图所示4个目录的模型,并将其放在插件的对应目录中。

图1 下载SDXL模型包并解压

模型包目录结构如下,将下载后的模型按照如下目录上传到对应文件夹中。

/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
support.huaweicloud.com/bestpractice-modelarts/modelarts_10_1503.html