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
"$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
echo "Downloading funasr..." &&
$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
else
echo "Skipping funasr download"
@ -40,7 +40,7 @@ fi
if [ "$USE_FASTERWHISPER" = "true" ]; then
echo "Downloading faster-whisper..." &&
$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
else
echo "Skipping faster-whisper download"
@ -49,9 +49,9 @@ fi
source "$HOME/anaconda3/etc/profile.d/conda.sh"
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
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
exit 1
fi

View File

@ -10,7 +10,7 @@ ARG CUDA_VERSION=12.4
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 \
build-essential \
gcc \

View File

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

View File

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