enginex-mlu370-diffusers/README.md

## Quickstart

### 构建镜像
```bash
docker build -t diffusers:v0.1 .
```
其中基础镜像 mlu370-pytorch:v25.01-torch2.5.0-torchmlu1.24.1-ubuntu22.04-py310 联系寒武纪厂商技术支持可获取

### 模型下载
模型地址：https://modelscope.cn/models/AI-ModelScope/stable-diffusion-v1-5
并放到目录：`/mnt/contest_ceph/zhanghao/models/stable-diffusion-v1-5`（如更改目录，请修改后面的执行脚本中的模型路径）

### 测试程序
1. 准备输入数据集，可以参考示例`dataset.json`
2. 在docker镜像里运行测试程序，会根据`dataset.json`内容，在`output`目录下生成图片文件。
```bash
./run_in_docker.sh
```

## 测试结果
|      | A100 平均生成时间（秒） | MLU370-x4 平均生成时间（秒） |  MLU370-x8 平均生成时间（秒） |
|------|-------------------------|----------------------------|----------------------------|
| 时间 | 1.5                  | 4.2                     | 5.3                     |