AI开发平台MODELARTS-模型适配:PyTorch模型转换为Onnx模型（可选）

时间：2025-02-25 19:54:36

AI开发平台MODELARTS 应用迁移

获取onnx模型有以下两种方式。下文介绍如何通过方式一进行操作。如果采用方式二，可以跳过此步骤。

方式一：使用官方提供的模型转换脚本将PyTorch模型转换为onnx模型。

方式二：对于提供了onnx模型的仓库，可以直接下载onnx模型。

通过git下载diffusers对应版本的源码。

git clone https://github.com/huggingface/diffusers.git -b v0.11.1

在diffusers的script/convert_stable_diffusion_checkpoint_to_onnx.py脚本中，可以通过执行以下命令生成onnx模型。其中，model_path指定PyTorch的模型根目录，output_path指定生成的onnx模型目录。
```
cd /home_host/work
python diffusers/scripts/convert_stable_diffusion_checkpoint_to_onnx.py --model_path "./runwayml/pytorch_models"  --output_path "./pytorch_to_onnx_models"
```