Files
enginex-bi_series-diffusers/run_in_docker.sh

4 lines
353 B
Bash
Raw Normal View History

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-28 17:52:17 +08:00
docker run -v `pwd`:/workspace -v /mnt/contest_ceph/zhanghao/models/stable-diffusion-v1-5:/workspace/stable-diffusion-v1-5 --device=dev/iluvatar1:/dev/iluvatar0 $image python3 main.py --model "./stable-diffusion-v1-5" --json "dataset.json" --results "results.json" --outdir "output" --device cuda --dtype fp16