From 004017c6110084aa2fa19919797b8d60421f261e Mon Sep 17 00:00:00 2001
From: XXXXRT666 <157766680+XXXXRT666@users.noreply.github.com>
Date: Sun, 4 May 2025 11:11:45 +0100
Subject: [PATCH] update docker_build

---
 docker_build.sh | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

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