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

Update install.sh

Browse Source
This commit is contained in:
XXXXRT666 2025-05-22 17:22:47 +03:00 committed by GitHub
parent 2a70ee3c85
commit 43de4e17d1
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
install.sh
View File

@ -28,7 +28,7 @@ print_help() {
echo "Usage: bash install.sh [OPTIONS]" echo "Usage: bash install.sh [OPTIONS]"
echo "" echo ""
echo "Options:" echo "Options:"
echo " --device CU124|CU128|ROCM|MPS|CPU Specify the Device (REQUIRED)" echo " --device CU126|CU128|ROCM|MPS|CPU Specify the Device (REQUIRED)"
echo " --source HF|HF-Mirror|ModelScope Specify the model source (REQUIRED)" echo " --source HF|HF-Mirror|ModelScope Specify the model source (REQUIRED)"
echo " --download-uvr5 Enable downloading the UVR5 model" echo " --download-uvr5 Enable downloading the UVR5 model"
echo " -h, --help Show this help message and exit" echo " -h, --help Show this help message and exit"
@ -68,8 +68,8 @@ while [[ $# -gt 0 ]]; do
;; ;;
--device) --device)
case "$2" in case "$2" in
CU124) CU126)
CUDA=124 CUDA=126
USE_CUDA=true USE_CUDA=true
;; ;;
CU128) CU128)
@ -87,7 +87,7 @@ while [[ $# -gt 0 ]]; do
;; ;;
*) *)
echo "Error: Invalid Device: $2" echo "Error: Invalid Device: $2"
echo "Choose From: [CU124, CU128, ROCM, MPS, CPU]" echo "Choose From: [CU126, CU128, ROCM, MPS, CPU]"
exit 1 exit 1
;; ;;
esac esac
@ -251,8 +251,8 @@ if [ "$USE_CUDA" = true ] && [ "$WORKFLOW" = false ]; then
echo "Installing PyTorch with CUDA support..." echo "Installing PyTorch with CUDA support..."
if [ "$CUDA" = 128 ]; then if [ "$CUDA" = 128 ]; then
pip install torch torchaudio --index-url https://download.pytorch.org/whl/cu128 pip install torch torchaudio --index-url https://download.pytorch.org/whl/cu128
elif [ "$CUDA" = 124 ]; then elif [ "$CUDA" = 126 ]; then
pip install torch==2.6 torchaudio --index-url https://download.pytorch.org/whl/cu124 pip install torch==2.6 torchaudio --index-url https://download.pytorch.org/whl/cu126
fi fi
elif [ "$USE_ROCM" = true ] && [ "$WORKFLOW" = false ]; then elif [ "$USE_ROCM" = true ] && [ "$WORKFLOW" = false ]; then
echo "Installing PyTorch with ROCm support..." echo "Installing PyTorch with ROCm support..."