enginex-c_series-diffusers/README.md

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, 下载地址:

• modelscope
• hugging face

测试程序

1. 准备输入数据集，可以参考示例dataset.json
2. 在docker镜像里运行测试程序，会根据dataset.json内容，在output目录下生成图片文件。
3. 示例使用 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