forked from EngineX-Ascend/enginex-ascend-910-diffusers
29 lines
832 B
Markdown
29 lines
832 B
Markdown
## Installation
|
||
参考Dockerfile，构建运行镜像
|
||
|
||
## Quickstart
|
||
|
||
### 测试程序
|
||
1. 下载模型：https://modelscope.cn/models/AI-ModelScope/stable-diffusion-v1-5
|
||
|
||
2. 运行测试程序
|
||
|
||
修改测试程序`test.py`里面的模型路径，直接执行即可
|
||
|
||
```bash
|
||
python3 test.py
|
||
```
|
||
|
||
### 批量测试程序
|
||
1. 准备输入数据集`dataset.json`，可以参考示例`dataset.json`
|
||
2. 运行测试程序
|
||
```bash
|
||
python3 main.py --model "/mnt/contest_ceph/zhanghao/models/stable-diffusion-v1-5" --json "dataset.json" --results "results.json" --outdir "output" --device cuda --dtype fp16
|
||
```
|
||
|
||
## 测试结果
|
||
| | A100 平均生成时间（秒） | 天垓100 平均生成时间（秒） |
|
||
|------|-------------------------|----------------------------|
|
||
| 时间 | 1.4289 | 7.5609 |
|
||
|