Stable-Diffusion-v1.5 示例

此文档讲述使用 QAI AppBuilder Python API 利用 Qualcomm® Hexagon™ Processor (NPU) 推理 Stable-Diffusion-v1.5 文生图模型。

示例支持设备

设备	SoC
Fogwise® AIRbox Q900	QCS9075

安装 QAI AppBuilder

提示

1. 请根据 QAI AppBuilder 安装方法 安装 QAI AppBuilder。

2. 请根据 创建 ADSP 环境变量 配置 ADSP 环境变量。

运行示例

安装依赖

在已激活的虚拟环境中安装示例依赖：

Device

pip3 install requests tqdm qai-hub py3-wget Pillow torch torchvision opencv-python-headless transformers diffusers

运行脚本

• 进入上游示例目录

  Device

  cd qai-appbuilder/samples

• 准备输入数据（示例自带输入文件时可直接运行；也可按脚本参数指定）

output image

• 执行推理

  Device

  python3 GenerativeAI/Image_Generation/stable_diffusion_v1_5/stable_diffusion_v1_5.py --chipset 9075 --prompt "a cat"

成功后会在示例目录下生成图片，例如：

GenerativeAI/Image_Generation/stable_diffusion_v1_5/images/<timestamp>_512.jpg

AIRbox Q900 上首次运行会下载 Text Encoder / UNet / VAE（合计约 1 GB 级），网络较慢时可能需数十分钟到数小时。

提示

依赖 transformers 与 diffusers。推荐直接运行脚本（避免 run_inference.py --args 对含空格 prompt 的引号问题）：

python3 GenerativeAI/Image_Generation/stable_diffusion_v1_5/stable_diffusion_v1_5.py --chipset 9075 --prompt "a cat"