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