Optimum提供与OpenVINO兼容的稳定扩散管道,可在各种英特尔处理器上执行推理
from optimum.intel import OVStableDiffusionPipeline
model_id = "runwayml/stable-diffusion-v1-5" pipeline = OVStableDiffusionPipeline.from_pretrained(model_id, export=True) prompt = "sailing ship in storm by Rembrandt" image = pipeline(prompt).images[0] image.save("open_vino.jpg") # 不要忘记保存输出的模型
pipeline.save_pretrained("openvino-sd-v1-5")