enginex-r_series-diffusers
运行于【昆仑芯 R200】系列算力卡的【文本生成图片】引擎,基于 diffusion 架构,支持 Stable Diffusion 等最新流行模型
Quickstart
构建镜像
需要xpytorch-cp38-torch201-ubuntu2004-x64-socket.run文件请前往此处下载
docker build -t r200_8f-diffuser:v0.21.4 .
其中基础镜像 r200-8f_xmlir-ubuntu_2004_x86_64:v0.27 通过联系昆仑芯厂商技术支持可获取
模型下载
模型地址:https://modelscope.cn/models/AI-ModelScope/stable-diffusion-v1-5
并放到目录:/mnt/data/models/stable-diffusion-v1-5(如更改目录,请修改后面的执行脚本中的模型路径)
测试程序
- 准备输入数据集,可以参考示例
dataset.json - 在docker镜像里运行测试程序,会根据
dataset.json内容,在output目录下生成图片文件。
./run_in_docker.sh
测试结果
stable-diffusion-v1-5模型
| A100 平均生成时间(秒) | R200_8F 平均生成时间(秒) | |
|---|---|---|
| 时间 | 3.0721 | 4.6615 |
Description
Languages
Python
86%
Shell
9.1%
Dockerfile
4.9%