AI开发平台MODELARTS-SDXL基于DevServer适配MindSpore-Lite NPU推理指导（6.3.T041）:Step4 模型转换

时间：2024-07-06 17:52:04

AI开发平台MODELARTS

Step4 模型转换

在进行模型转换前，需先设置环境变量。

cd /home/ma-user/sdxl_ms
export LITE_HOME=/usr/local/mindspore-lite/mindspore-lite-2.3.0rc1-linux-aarch64
export LD_LIBRARY_PATH=$LITE_HOME/runtime/lib:$LITE_HOME/runtime/third_party/dnnl:$LITE_HOME/tools/converter/lib:$LD_LIBRARY_PATH
export PATH=$LITE_HOME/tools/converter/converter:$LITE_HOME/tools/benchmark:$PATH

修改模型转换脚本。
模型转换的脚本是/home/ma-user/sdxl_ms/run_onnx_to_mindir.sh，脚本内容如下，请根据实际情况修改。

图4 修改模型转换脚本

这里提供的模型适配包支持5种分档位图片大小，分别是512x512 768x768 960x960 1024x1024 1280x1280
执行如下命令转换模型。
```
sh run_onnx_to_mindir.sh
```
出现CONVERT RESULT SUC CES S:0即为转模型成功，这里会自动依次转换5个模型。转换时间较长（大约35min），请耐心等待。