在1684X SOC 一键部署Stable diffusion 模型

文章目录

    • 一. 准备工作
    • 二. 一键部署
    • 三. 预期结果

一. 准备工作

  1. 准备一块1684XSOC开发板,将其调为soc模式,使用最新的sdcard 刷机包,刷机包下载地址:

https://developer.sophgo.com/site/index/material/29/all.html

特别注意:
模型需要使用sail包,但是因为官网的刷机包版本可能会更新,而sail包需要与libsophon, sophonmw的版本匹配,所以需要自己编译sail包,编译方法可以参考sail的readme文档。也可以向算能公司官方人员申请编译好的sail包。

  1. 检查环境
  • 检查是否存在sophon环境 (ls /opt/sophon)

在这里插入图片描述

  • 查看当前版本 bm_version

在1684X SOC 一键部署Stable diffusion 模型_第1张图片

  • 查看显存大小 bm-smi

  • 查看是否正确安装sail

python3 -c "import sophon.sail as sail; print(sail.__version__)"

二. 一键部署

  1. 下载代码和模型文件

/data 目录下新建一个文件夹,例如stable_diffusion

安装 pip3 install dfn

下载脚本: python3 -m dfn --url http://219.142.246.77:65000/sharing/sgZNjWcYK

解压后会得到一个stable_diffusion文件夹,里面包含了

在1684X SOC 一键部署Stable diffusion 模型_第2张图片

安装依赖 pip3 install -r requirements.txt

  1. 运行脚本

bash setup.sh

通过web.log 日志可以streamlit开启的端口,通过浏览器访问即可。

三. 预期结果

在浏览器中输入网址后出现如下界面,说明部署成功。

在1684X SOC 一键部署Stable diffusion 模型_第3张图片

你可能感兴趣的:(TPU实战课,stable,diffusion,模型部署,人工智能,算法)