blackteay/GPT-SoVITS
1
0
Fork 0
mirror of https://github.com/RVC-Boss/GPT-SoVITS.git synced 2025-08-25 06:45:29 +08:00
Code Issues Packages Projects Releases Wiki Activity

update docker_build

Browse Source
This commit is contained in:
XXXXRT666 2025-05-04 11:11:45 +01:00
parent 9c42a133d3
commit 004017c611
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
docker_build.sh
View File

@ -68,15 +68,15 @@ done
TARGETPLATFORM=$(uname -m | grep -q 'x86' && echo "linux/amd64" || echo "linux/arm64") TARGETPLATFORM=$(uname -m | grep -q 'x86' && echo "linux/amd64" || echo "linux/arm64")
if [ $LITE = true ]; then if [ $LITE = true ]; then
CUDA_BASE="runtime" TORCH_BASE="lite"
else else
CUDA_BASE="devel" TORCH_BASE="full"
fi fi
docker build \ docker build \
--build-arg CUDA_VERSION=$CUDA_VERSION \ --build-arg CUDA_VERSION=$CUDA_VERSION \
--build-arg LITE=$LITE \ --build-arg LITE=$LITE \
--build-arg TARGETPLATFORM="$TARGETPLATFORM" \ --build-arg TARGETPLATFORM="$TARGETPLATFORM" \
--build-arg CUDA_BASE=$CUDA_BASE \ --build-arg TORCH_BASE=$TORCH_BASE \
-t "${USER}/gpt-sovits:local" \ -t "${USER}/gpt-sovits:local" \
. .