diff --git a/.github/workflows/docker-publish.yaml b/.github/workflows/docker-publish.yaml
index 1f6aa35f..20dcea89 100644
--- a/.github/workflows/docker-publish.yaml
+++ b/.github/workflows/docker-publish.yaml
@@ -26,16 +26,16 @@ jobs:
       matrix:
         include:
           - cuda_version: 12.4
-            lite: false
+            lite: true
             tag_prefix: lite-cu124
           - cuda_version: 12.4
-            lite: true
+            lite: false
             tag_prefix: cu124
           - cuda_version: 12.8
-            lite: false
+            lite: true
             tag_prefix: lite-cu128
           - cuda_version: 12.8
-            lite: true
+            lite: false
             tag_prefix: cu128
 
     steps:
diff --git a/Docker/setup.sh b/Docker/setup.sh
index 55ef884c..14f4d8f2 100644
--- a/Docker/setup.sh
+++ b/Docker/setup.sh
@@ -50,9 +50,7 @@ source "$HOME/anaconda3/etc/profile.d/conda.sh"
 
 conda config --add channels conda-forge
 
-conda config --set channel_priority strict
-
-conda install -n base -c conda-forge conda=25.3.1
+conda update --all -y
 
 if [ "$LITE" = "true" ]; then
     bash install.sh --device "CU${CUDA_VERSION//./}" --source HF
diff --git a/docker_build.sh b/docker_build.sh
index 28e731ae..ea4960cd 100644
--- a/docker_build.sh
+++ b/docker_build.sh
@@ -65,8 +65,11 @@ while [[ $# -gt 0 ]]; do
     esac
 done
 
+TARGETPLATFORM=$(uname -m | grep -q 'x86_64' && echo "linux/amd64" || echo "linux/arm64")
+
 docker build \
     --build-arg CUDA_VERSION=$CUDA_VERSION \
     --build-arg LITE=$LITE \
-    -t "${USER}/gpt-sovits:local" \
+    --build-arg TARGETPLATFORM="$TARGETPLATFORM"
+-t "${USER}/gpt-sovits:local" \
     .