init commit

This commit is contained in:
2025-08-20 17:53:15 +08:00
commit e234ac5489
6 changed files with 230 additions and 0 deletions

7
test.sh Executable file
View File

@@ -0,0 +1,7 @@
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