前言

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-inferencetops-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