Quickstart

构建镜像

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目录下生成图片文件。

./run_in_docker.sh

测试结果

	A100 平均生成时间（秒）	MLU370-x4 平均生成时间（秒）	MLU370-x8 平均生成时间（秒）
时间	1.5	4.2	5.3