blackteay/GPT-SoVITS
1
0
Fork 0
mirror of https://github.com/RVC-Boss/GPT-SoVITS.git synced 2025-08-31 19:43:09 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update docker_build.sh

Browse Source
This commit is contained in:
XXXXRT666 2025-05-22 17:27:43 +03:00 committed by GitHub
parent 8ddd3667d0
commit 825c689940
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
docker_build.sh
View File

@ -14,18 +14,18 @@ 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
LITE=false LITE=false
CUDA_VERSION=12.4 CUDA_VERSION=12.6
print_help() { print_help() {
echo "Usage: bash docker_build.sh [OPTIONS]" echo "Usage: bash docker_build.sh [OPTIONS]"
echo "" echo ""
echo "Options:" echo "Options:"
echo " --cuda 12.4|12.8 Specify the CUDA VERSION (REQUIRED)" echo " --cuda 12.6|12.8 Specify the CUDA VERSION (REQUIRED)"
echo " --lite Build a Lite Image" echo " --lite Build a Lite Image"
echo " -h, --help Show this help message and exit" echo " -h, --help Show this help message and exit"
echo "" echo ""
echo "Examples:" echo "Examples:"
echo " bash docker_build.sh --cuda 12.4 --funasr --faster-whisper" echo " bash docker_build.sh --cuda 12.6 --funasr --faster-whisper"
} }
# Show help if no arguments provided # Show help if no arguments provided
@ -39,15 +39,15 @@ while [[ $# -gt 0 ]]; do
case "$1" in case "$1" in
--cuda) --cuda)
case "$2" in case "$2" in
12.4) 12.6)
CUDA_VERSION=12.4 CUDA_VERSION=12.6
;; ;;
12.8) 12.8)
CUDA_VERSION=12.8 CUDA_VERSION=12.8
;; ;;
*) *)
echo "Error: Invalid CUDA_VERSION: $2" echo "Error: Invalid CUDA_VERSION: $2"
echo "Choose From: [12.4, 12.8]" echo "Choose From: [12.6, 12.8]"
exit 1 exit 1
;; ;;
esac esac