814 B
814 B
Quickstart
构建镜像
docker build -t diffusers:v0.1 .
模型下载
模型地址：https://modelscope.cn/models/AI-ModelScope/stable-diffusion-v1-5
并放到目录：/mnt/contest_ceph/zhanghao/models/stable-diffusion-v1-5（如更改目录，请修改后面的执行脚本中的模型路径）
测试程序
- 准备输入数据集，可以参考示例
dataset.json - 在docker镜像里运行测试程序，会根据
dataset.json内容，在output目录下生成图片文件。
./run_in_docker.sh
测试结果
stable-diffusion-v1-5模型
| A100 平均生成时间（秒） | 昇腾910B 平均生成时间（秒） | |
|---|---|---|
| 时间 | 1.6208 | 7.2901 |