This commit is contained in:
2025-09-15 16:52:26 +08:00
parent 86d07c55e5
commit c77b95a340
5 changed files with 246 additions and 1 deletions

View File

@@ -1,3 +1,30 @@
# enginex-r_series-diffusers
运行于【昆仑芯 R200】系列算力卡的【文本生成图片】引擎基于 diffusion 架构,支持 Stable Diffusion 等最新流行模型
运行于【昆仑芯 R200】系列算力卡的【文本生成图片】引擎基于 diffusion 架构,支持 Stable Diffusion 等最新流行模型
## Quickstart
### 构建镜像
需要xpytorch-cp38-torch201-ubuntu2004-x64-socket.run文件请前往[此处](https://git.modelhub.org.cn:9443/EngineX-Kunlunxin/r200_8f_xpytorch)下载
```bash
docker build -t r200_8f-diffuser:v0.21.4 .
```
### 模型下载
模型地址https://modelscope.cn/models/AI-ModelScope/stable-diffusion-v1-5
并放到目录:`/mnt/data/models/stable-diffusion-v1-5`(如更改目录,请修改后面的执行脚本中的模型路径)
### 测试程序
1. 准备输入数据集,可以参考示例`dataset.json`
2. 在docker镜像里运行测试程序会根据`dataset.json`内容,在`output`目录下生成图片文件。
```bash
./run_in_docker.sh
```
## 测试结果
### stable-diffusion-v1-5模型
| | A100 平均生成时间(秒) | R200_8F 平均生成时间(秒) |
|------|-------------------------|----------------------------|
| 时间 | 3.0721 | 4.6615 |