From 455848137a2f67345e08d93eee879b3db46c757f Mon Sep 17 00:00:00 2001 From: Zhang Hao Date: Thu, 28 Aug 2025 17:51:11 +0800 Subject: [PATCH] mount current dir to workspace --- run_in_docker.sh | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/run_in_docker.sh b/run_in_docker.sh index 0767548..fed2bc0 100755 --- a/run_in_docker.sh +++ b/run_in_docker.sh @@ -1,3 +1,3 @@ #! /usr/bin/env 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 \ No newline at end of file +docker run -v `pwd`:/workspace -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