diff --git a/docker_folder/model_dir/.gitignore b/docker_folder/model_dir/.gitignore new file mode 100644 index 00000000..a5baada1 --- /dev/null +++ b/docker_folder/model_dir/.gitignore @@ -0,0 +1,3 @@ +* +!.gitignore + diff --git a/docker_folder/pretrain/.gitignore b/docker_folder/pretrain/.gitignore new file mode 100644 index 00000000..a5baada1 --- /dev/null +++ b/docker_folder/pretrain/.gitignore @@ -0,0 +1,3 @@ +* +!.gitignore + diff --git a/start_docker.sh b/start_docker.sh index 4a6506e7..4e59bb81 100644 --- a/start_docker.sh +++ b/start_docker.sh @@ -24,6 +24,8 @@ if [ "${USE_GPU}" = "on" ]; then -e EX_PORT=${EX_PORT} \ -e LOCAL_UID=$(id -u $USER) \ -e LOCAL_GID=$(id -g $USER) \ + -v `pwd`/docker_folder/model_dir:/voice-changer/server/model_dir + -v `pwd`/docker_folder/pretrain:/voice-changer/server/pretrain -p ${EX_PORT}:18888 \ $DOCKER_IMAGE -p 18888 --https true \ --content_vec_500 pretrain/checkpoint_best_legacy_500.pt \