add input config

This commit is contained in:
2025-08-28 12:49:39 +08:00
parent cdefc1873e
commit 33c57acb1b
2 changed files with 19 additions and 6 deletions

View File

@@ -1 +1 @@
python3 main.py --model "/mnt/contest_ceph/zhanghao/models/stable-diffusion-v1-5" --json "dataset.json" --results "results.json" --outdir "output" --device cuda --dtype fp16
python3 main.py --model "/mnt/contest_ceph/zhanghao/models/stable-diffusion-v1-5" --json "dataset.json" --results "results.json" --outdir "output" --device cuda --dtype fp16 --num_inference_steps 20