AI开发平台MODELARTS-pipeline代码适配:修改代码依赖

时间:2024-12-09 20:36:14

修改代码依赖

新建并进入/home_host/work/pipeline目录。

mkdir -p /home_host/work/pipeline
cd /home_host/work/pipeline

将onnx pipeline依赖的图生图源码“pipeline_onnx_stable_diffusion_img2img.py”复制到该目录下,名称改为“pipeline_onnx_stable_diffusion_img2img_mslite.py”,以便与源文件名称区分。但是这样也会导致无法正确找到源码中相对路径下的依赖,需要将对于diffusers包内的相对路径修改为绝对路径的形式。

图1 代码依赖修改前与修改后

将推理代码“modelarts-ascend/examples/AIGC/stable_diffusion/onnx_pipeline.py”也复制一份到该目录,名称改为“mslite_pipeline.py”,迁移后的推理代码中的pipeline需要修改为从复制的onnx pipeline文件导入:

# onnx_pipeline.py
from pipeline_onnx_stable_diffusion_img2img_mslite import OnnxStableDiffusionImg2ImgPipeline
support.huaweicloud.com/bestpractice-modelarts/modelarts_10_2005.html