前言¶
stable_diffusion_gcu是适配于Enflame i20 gcu的stable diffusion类模型的推理库,基于ONNX/TopsInference进行推理。
版本信息¶
日期 | 版本 | 作者 | 新增功能 |
---|---|---|---|
20230704 | v1.0 | Enflame | |
20231220 | v1.3 | Enflame |
原理及功能介绍¶
当前支持的功能包括:
StableDiffusion V1.5
的文生图、图生图、融合lora
推理功能StableDiffusion V1.5 controlnet
共10种可控生成模型的文生图、图生图、融合lora
推理功能StableDiffusion V2
的文生图、图生图推理功能StableDiffusion V2.1
的文生图、图生图推理功能StableDiffusion XL 1.0
的文生图,图生图、融合lora
推理功能StableDiffusion XL 1.0
的6种可控生成(T2I-Adapter)
模型推理功能基于给定图片的标签反推
(DeepDanBooru)
的推理功能图像超分辨率重建
(UpScaler)
的推理功能
安装使用说明¶
安装
tops-inference
和tops-sdk
使用命令 `dpkg -l | grep tops` 查看tops-inference和tops-sdk是否安装成功
安装
stable_diffusion_gcu
python3.10: python3.10 -m pip install stable_diffusion_gcu-2.5.1-py3.10-none-any.whl --extra-index-url https://download.pytorch.org/whl/cpu --force-reinstall