Quickstart
Docker Image
自行联系沐曦官方获取下载链接
docker pull cr.metax-tech.com/public-ai-release/maca/diffusers.training:maca.ai3.0.0.5-torch2.4-py310-ubuntu22.04-amd64
模型
stable-diffusion-v1-5, 下载地址:
测试程序
- 准备输入数据集，可以参考示例
dataset.json - 在docker镜像里运行测试程序，会根据
dataset.json内容，在output目录下生成图片文件。 - 示例使用 metax-docker 做为演示 docker runtime, 如用原生 docker 需自行挂载 GPU 等设备到容器
完整示例
git clone https://git.modelhub.org.cn:9443/EngineX-MetaX/enginex-c_series-diffusers
docker pull cr.metax-tech.com/public-ai-release/maca/diffusers.training:maca.ai3.0.0.5-torch2.4-py310-ubuntu22.04-amd64
cd enginex-c_series-diffusers/
# git clone 先确认 git-lfs 已安装
git lfs install
git clone https://www.modelscope.cn/AI-ModelScope/stable-diffusion-v1-5.git
metax-docker run --gpus=[0] -it --entrypoint /bin/bash -v $(realpath .):/workspace/app -v $(realpath stable-diffusion-v1-5):/model --entrypoint /bin/bash cr.metax-tech.com/public-ai-release/maca/diffusers.training:maca.ai3.0.0.5-torch2.4-py310-ubuntu22.04-amd64
#> inside container
cd app/
python3 main.py \
--model "/model" \
--json "dataset.json" \
--results "results.json" \
--outdir "output" \
--device cuda \
--dtype fp16
测试结果
| A100 平均生成时间（秒） | MetaX C500 平均生成时间（秒） | |
|---|---|---|
| 时间 | 2.2142 | 4.5260 |