add README

2025-08-25 14:14:19 +08:00
parent 931a35a7db
commit 4d35251531
3 changed files with 35 additions and 6 deletions
--- a/README.md
+++ b/README.md
@@ -0,0 +1,28 @@
+## 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                     |
+