diff --git a/build_docker.sh b/build_docker.sh
index ac0d2b9..42200d5 100644
--- a/build_docker.sh
+++ b/build_docker.sh
@@ -15,7 +15,7 @@ docker buildx build \
 --label org.opencontainers.image.source=$GIT_REPO \
 --label org.opencontainers.image.version=$GIT_TAG \
 --build-arg version=$GIT_TAG \
---tag lama-cleaner:cpu-$GIT_TAG .
+--tag cwq1913/lama-cleaner:cpu-$GIT_TAG .
 
 
 echo "Building NVIDIA GPU docker image..."
@@ -28,4 +28,4 @@ docker buildx build \
 --label org.opencontainers.image.source=$GIT_REPO \
 --label org.opencontainers.image.version=$GIT_TAG \
 --build-arg version=$GIT_TAG \
---tag lama-cleaner:gpu-$GIT_TAG .
+--tag cwq1913/lama-cleaner:gpu-$GIT_TAG .
diff --git a/docker/CPUDockerfile b/docker/CPUDockerfile
index 33bb7bb..9304267 100644
--- a/docker/CPUDockerfile
+++ b/docker/CPUDockerfile
@@ -6,7 +6,7 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
     curl
 
 RUN pip install --upgrade pip && \
-    pip install torch==1.12.1 --extra-index-url https://download.pytorch.org/whl/cpu
+    pip install torch==1.13.1 --extra-index-url https://download.pytorch.org/whl/cpu
 
 ARG version
 
diff --git a/docker/GPUDockerfile b/docker/GPUDockerfile
index f6b9264..8b5b96d 100644
--- a/docker/GPUDockerfile
+++ b/docker/GPUDockerfile
@@ -1,4 +1,4 @@
-FROM nvidia/cuda:11.6.1-runtime-ubuntu20.04
+FROM nvidia/cuda:11.7.1-runtime-ubuntu20.04
 
 RUN apt-get update && apt-get install -y --no-install-recommends \
     software-properties-common \
@@ -6,8 +6,8 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
     curl python3-pip
 
 RUN pip3 install --upgrade pip && \
-    pip3 install torch==1.12.1+cu116 --extra-index-url https://download.pytorch.org/whl/cu116
-    
+    pip3 install torch==1.13.1 xformers==0.0.16rc425
+
 ARG version 
 
 RUN pip3 install lama-cleaner==$version