sunjichen/enginex-ascend-910-diffusers
1
0
Fork 0
forked from EngineX-Ascend/enginex-ascend-910-diffusers
Code Pull Requests Activity

add README

Browse Source
This commit is contained in:
Zhang Hao
2025-08-25 14:14:19 +08:00
parent 931a35a7db
commit 4d35251531
3 changed files with 35 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
Dockerfile
View File

@@ -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/

1
Dockerfile Symbolic link
View File

@@ -0,0 +1 @@
Dockerfile.bi100

6
Dockerfile.bi100 Normal file
View File

@@ -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/

28
README.md Normal file
View File

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