Installation

参考Dockerfile，构建运行镜像

Quickstart

测试程序

1. 下载模型：https://modelscope.cn/models/AI-ModelScope/stable-diffusion-v1-5

2. 运行测试程序

修改测试程序test.py里面的模型路径，直接执行即可

python3 test.py

批量测试程序

1. 准备输入数据集dataset.json，可以参考示例dataset.json
2. 运行测试程序

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