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