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 |
|
|||
|
|
|