diff --git a/Dockerfile b/Dockerfile index aa0b7fd7..14e9ef39 100644 --- a/Dockerfile +++ b/Dockerfile @@ -28,7 +28,8 @@ WORKDIR /workspace/GPT-SoVITS COPY . /workspace/GPT-SoVITS -ARG WGET_CMD +ARG WGET_CMD=wget --tries=25 --wait=5 --read-timeout=40 --retry-on-http-error=404 +ENV WGET_CMD=${WGET_CMD} RUN eval "$WGET_CMD -O anaconda.sh https://repo.anaconda.com/archive/Anaconda3-2024.10-1-Linux-x86_64.sh" && \ bash anaconda.sh -b -p /root/anaconda3 && \ diff --git a/docker_build.sh b/docker_build.sh index 6f13cd4f..212c8d1b 100644 --- a/docker_build.sh +++ b/docker_build.sh @@ -16,7 +16,6 @@ trap 'echo "Error Occured at \"$BASH_COMMAND\" with exit code $?"; exit 1' ERR USE_FUNASR=false USE_FASTERWHISPER=false CUDA_VERSION=12.4 -WGET_CMD="wget --tries=25 --wait=5 --read-timeout=40 --retry-on-http-error=404" print_help() { echo "Usage: bash docker_build.sh [OPTIONS]" @@ -79,6 +78,5 @@ docker build \ --build-arg CUDA_VERSION=$CUDA_VERSION \ --build-arg USE_FUNASR=$USE_FUNASR \ --build-arg USE_FASTERWHISPER=$USE_FASTERWHISPER \ - --build-arg WGET_CMD="$WGET_CMD" \ -t "${USER}/gpt-sovits:local" \ .