2025-09-03 10:16:24 +08:00
2025-08-20 17:53:15 +08:00
2025-08-25 14:14:19 +08:00
2025-08-28 11:17:44 +08:00
2025-09-03 10:16:24 +08:00
2025-08-25 14:14:19 +08:00
2025-08-26 17:25:25 +08:00
2025-08-26 18:13:06 +08:00
2025-09-03 10:16:24 +08:00
2025-08-26 18:13:06 +08:00
2025-08-21 11:00:53 +08:00
2025-09-03 10:16:24 +08:00
2025-08-26 18:13:06 +08:00
2025-08-26 18:13:06 +08:00
2025-09-03 10:16:24 +08:00
2025-08-28 12:49:39 +08:00

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(如更改目录,请修改后面的执行脚本中的模型路径)

测试程序

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

测试结果

A100 平均生成时间(秒) 天垓100 平均生成时间(秒)
时间 1.4289 7.5609
Description
运行于【昇腾-910B4】系列算力卡的【文本生成图片】引擎,基于 diffusion 架构,支持 Stable Diffusion 等最新流行模型
Readme 50 KiB
Languages
Python 90.1%
Shell 8.1%
Dockerfile 1.8%