forked from EngineX-Ascend/enginex-ascend-910-diffusers
26 lines
814 B
Markdown
26 lines
814 B
Markdown
## Quickstart
|
||
|
||
### 构建镜像
|
||
```bash
|
||
docker build -t diffusers:v0.1 .
|
||
```
|
||
|
||
### 模型下载
|
||
模型地址：https://modelscope.cn/models/AI-ModelScope/stable-diffusion-v1-5
|
||
并放到目录：`/mnt/contest_ceph/zhanghao/models/stable-diffusion-v1-5`（如更改目录，请修改后面的执行脚本中的模型路径）
|
||
|
||
### 测试程序
|
||
1. 准备输入数据集，可以参考示例`dataset.json`
|
||
2. 在docker镜像里运行测试程序，会根据`dataset.json`内容，在`output`目录下生成图片文件。
|
||
```bash
|
||
./run_in_docker.sh
|
||
```
|
||
|
||
## 测试结果
|
||
### stable-diffusion-v1-5模型
|
||
|
||
| | A100 平均生成时间（秒） | 昇腾910B 平均生成时间（秒） |
|
||
|------|-------------------------|----------------------------|
|
||
| 时间 | 1.6208 | 7.2901 |
|
||
|