diff --git a/test.sh b/test.sh index a3ec501..7bad55f 100755 --- a/test.sh +++ b/test.sh @@ -1,7 +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 \ No newline at end of file +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