This commit is contained in:
XXXXRT666 2025-04-30 21:49:14 +01:00
parent b11931c1bd
commit 1d2c6c5cbb
5 changed files with 16 additions and 21 deletions

View File

@ -41,4 +41,6 @@ rm anaconda.sh
rm $LOG_PATH rm $LOG_PATH
"$HOME/anaconda3/bin/conda" clean -p rm -rf "$HOME/anaconda3/pkgs"
mkdir "$HOME/anaconda3/pkgs"

View File

@ -31,7 +31,7 @@ fi
if [ "$USE_FUNASR" = "true" ]; then if [ "$USE_FUNASR" = "true" ]; then
echo "Downloading funasr..." && echo "Downloading funasr..." &&
$WGET_CMD "https://huggingface.co/XXXXRT/GPT-SoVITS-Pretrained/resolve/main/funasr.zip" && $WGET_CMD "https://huggingface.co/XXXXRT/GPT-SoVITS-Pretrained/resolve/main/funasr.zip" &&
unzip funasr.zip -d tools/asr/models/ && unzip -q funasr.zip -d tools/asr/models/ &&
rm -rf funasr.zip rm -rf funasr.zip
else else
echo "Skipping funasr download" echo "Skipping funasr download"
@ -40,7 +40,7 @@ fi
if [ "$USE_FASTERWHISPER" = "true" ]; then if [ "$USE_FASTERWHISPER" = "true" ]; then
echo "Downloading faster-whisper..." && echo "Downloading faster-whisper..." &&
$WGET_CMD "https://huggingface.co/XXXXRT/GPT-SoVITS-Pretrained/resolve/main/faster-whisper.zip" && $WGET_CMD "https://huggingface.co/XXXXRT/GPT-SoVITS-Pretrained/resolve/main/faster-whisper.zip" &&
unzip faster-whisper.zip -d tools/asr/models/ && unzip -q faster-whisper.zip -d tools/asr/models/ &&
rm -rf faster-whisper.zip rm -rf faster-whisper.zip
else else
echo "Skipping faster-whisper download" echo "Skipping faster-whisper download"
@ -49,9 +49,9 @@ fi
source "$HOME/anaconda3/etc/profile.d/conda.sh" source "$HOME/anaconda3/etc/profile.d/conda.sh"
if [ "$LITE" = "true" ]; then if [ "$LITE" = "true" ]; then
bash install.sh --device "CU${CUDA_VERSION//./}" --source HF --skip-check "$WORKFLOW" bash install.sh --device "CU${CUDA_VERSION//./}" --source HF
elif [ "$LITE" = "false" ]; then elif [ "$LITE" = "false" ]; then
bash install.sh --device "CU${CUDA_VERSION//./}" --source HF --skip-check "$WORKFLOW" --download-uvr5 bash install.sh --device "CU${CUDA_VERSION//./}" --source HF --download-uvr5
else else
exit 1 exit 1
fi fi

View File

@ -10,7 +10,7 @@ ARG CUDA_VERSION=12.4
ENV CUDA_VERSION=${CUDA_VERSION} ENV CUDA_VERSION=${CUDA_VERSION}
RUN DEBIAN_FRONTEND=noninteractive apt-get update && \ RUN DEBIAN_FRONTEND=noninteractive apt-get update -qq && \
apt-get install -y -qq --no-install-recommends \ apt-get install -y -qq --no-install-recommends \
build-essential \ build-essential \
gcc \ gcc \

View File

@ -13,8 +13,7 @@ fi
trap 'echo "Error Occured at \"$BASH_COMMAND\" with exit code $?"; exit 1' ERR trap 'echo "Error Occured at \"$BASH_COMMAND\" with exit code $?"; exit 1' ERR
USE_FUNASR=false LITE=false
USE_FASTERWHISPER=false
CUDA_VERSION=12.4 CUDA_VERSION=12.4
print_help() { print_help() {
@ -22,8 +21,7 @@ print_help() {
echo "" echo ""
echo "Options:" echo "Options:"
echo " --cuda 12.4|12.8 Specify the CUDA VERSION (REQUIRED)" echo " --cuda 12.4|12.8 Specify the CUDA VERSION (REQUIRED)"
echo " --funasr Build with FunASR Paraformer Model" echo " --lite Build a Lite Image"
echo " --faster-whisper Build with Faster-Whisper-Large-V3 Model"
echo " -h, --help Show this help message and exit" echo " -h, --help Show this help message and exit"
echo "" echo ""
echo "Examples:" echo "Examples:"
@ -55,12 +53,8 @@ while [[ $# -gt 0 ]]; do
esac esac
shift 2 shift 2
;; ;;
--funasr) --lite)
USE_FUNASR=true LITE=true
shift
;;
--faster-whisper)
USE_FASTERWHISPER=true
shift shift
;; ;;
*) *)
@ -73,7 +67,6 @@ done
docker build \ docker build \
--build-arg CUDA_VERSION=$CUDA_VERSION \ --build-arg CUDA_VERSION=$CUDA_VERSION \
--build-arg USE_FUNASR=$USE_FUNASR \ --build-arg LITE=$LITE \
--build-arg USE_FASTERWHISPER=$USE_FASTERWHISPER \
-t "${USER}/gpt-sovits:local" \ -t "${USER}/gpt-sovits:local" \
. .

View File

@ -162,7 +162,7 @@ else
echo "Download Pretrained Models" echo "Download Pretrained Models"
$WGET_CMD "$PRETRINED_URL" $WGET_CMD "$PRETRINED_URL"
unzip pretrained_models.zip unzip -q pretrained_models.zip
rm -rf pretrained_models.zip rm -rf pretrained_models.zip
mv pretrained_models/* GPT_SoVITS/pretrained_models mv pretrained_models/* GPT_SoVITS/pretrained_models
rm -rf pretrained_models rm -rf pretrained_models
@ -172,7 +172,7 @@ if [ ! -d "GPT_SoVITS/text/G2PWModel" ]; then
echo "Download G2PWModel" echo "Download G2PWModel"
$WGET_CMD "$G2PW_URL" $WGET_CMD "$G2PW_URL"
unzip G2PWModel.zip unzip -q G2PWModel.zip
rm -rf G2PWModel.zip rm -rf G2PWModel.zip
mv G2PWModel GPT_SoVITS/text/G2PWModel mv G2PWModel GPT_SoVITS/text/G2PWModel
else else
@ -186,7 +186,7 @@ if [ "$DOWNLOAD_UVR5" = "true" ]; then
echo "Download UVR5 Model" echo "Download UVR5 Model"
$WGET_CMD "$UVR5_URL" $WGET_CMD "$UVR5_URL"
unzip uvr5_weights.zip unzip -q uvr5_weights.zip
rm -rf uvr5_weights.zip rm -rf uvr5_weights.zip
mv uvr5_weights/* tools/uvr5/uvr5_weights mv uvr5_weights/* tools/uvr5/uvr5_weights
rm -rf uvr5_weights rm -rf uvr5_weights