2025-08-20 17:53:15 +08:00
|
|
|
#! /usr/bin/env bash
|
2025-08-26 18:13:06 +08:00
|
|
|
image=diffusers:v0.1
|
2025-08-26 21:46:07 +08:00
|
|
|
device=cambricon_dev1
|
|
|
|
|
docker run -v /mnt/contest_ceph/zhanghao/models/stable-diffusion-v1-5:/workspace/stable-diffusion-v1-5 --device=/dev/$device:/dev/cambricon_dev0 --device=/dev/cambricon_ctl:/dev/cambricon_ctl $image python3 main.py --model "./stable-diffusion-v1-5" --json "dataset.json" --results "results.json" --outdir "output" --device cuda --dtype fp16
|
