update readme

This commit is contained in:
2025-08-26 18:13:06 +08:00
parent fcf7f30797
commit be947dfc79
4 changed files with 15 additions and 19 deletions

View File

@@ -1,3 +1,3 @@
#! /usr/bin/env bash
image=harbor-contest.4pd.io/zhanghao/diffusers:bi100-0.2
docker run -it -v /root/zhanghao:/workspace -v /mnt:/mnt --device=dev/iluvatar1:/dev/iluvatar0 $image bash
image=diffusers:v0.1
docker run -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