From 4d35251531c1bbf31bf65e699e8a73ea11ad42e9 Mon Sep 17 00:00:00 2001 From: Zhang Hao Date: Mon, 25 Aug 2025 14:14:19 +0800 Subject: [PATCH] add README --- Dockerfile | 7 +------ Dockerfile.bi100 | 6 ++++++ README.md | 28 ++++++++++++++++++++++++++++ 3 files changed, 35 insertions(+), 6 deletions(-) mode change 100644 => 120000 Dockerfile create mode 100644 Dockerfile.bi100 create mode 100644 README.md diff --git a/Dockerfile b/Dockerfile deleted file mode 100644 index 364c3c9..0000000 --- a/Dockerfile +++ /dev/null @@ -1,6 +0,0 @@ -FROM git.modelhub.org.cn:980/enginex-iluvatar/bi100-3.2.1-x86-ubuntu20.04-py3.10-poc-llm-infer:v1.2.2 - -WORKDIR /workspace -ENV PT_SDPA_ENABLE_HEAD_DIM_PADDING=1 -RUN pip install diffusers==0.34.0 -COPY main.py test.sh dataset.json /workspace/ diff --git a/Dockerfile b/Dockerfile new file mode 120000 index 0000000..45d555f --- /dev/null +++ b/Dockerfile @@ -0,0 +1 @@ +Dockerfile.bi100 \ No newline at end of file diff --git a/Dockerfile.bi100 b/Dockerfile.bi100 new file mode 100644 index 0000000..364c3c9 --- /dev/null +++ b/Dockerfile.bi100 @@ -0,0 +1,6 @@ +FROM git.modelhub.org.cn:980/enginex-iluvatar/bi100-3.2.1-x86-ubuntu20.04-py3.10-poc-llm-infer:v1.2.2 + +WORKDIR /workspace +ENV PT_SDPA_ENABLE_HEAD_DIM_PADDING=1 +RUN pip install diffusers==0.34.0 +COPY main.py test.sh dataset.json /workspace/ diff --git a/README.md b/README.md new file mode 100644 index 0000000..0f1e953 --- /dev/null +++ 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 | +