Installation
参考Dockerfile，构建运行镜像
Quickstart
测试程序
修改测试程序test.py里面的模型路径，直接执行即可
python3 test.py
批量测试程序
- 准备输入数据集
dataset.json，可以参考示例dataset.json - 运行测试程序
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.5 | 4.1 |