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