diff --git a/setup_files/Dockerfile b/setup_files/Dockerfile
index d9bb73c..7657b5e 100644
--- a/setup_files/Dockerfile
+++ b/setup_files/Dockerfile
@@ -1,4 +1,4 @@
-FROM python:3.12.0
+FROM python:3.11.3-slim-bullseye
 COPY ./requirements.txt /
 RUN apt-get update \
     && apt-get install wget libgl1-mesa-glx libglib2.0-0 --no-install-recommends -y \
diff --git a/setup_files/requirements.txt b/setup_files/requirements.txt
index 7ecba63..7e1e17b 100644
--- a/setup_files/requirements.txt
+++ b/setup_files/requirements.txt
@@ -3,9 +3,9 @@ accelerate
 diffusers[torch]==0.21.4
 onnxruntime==1.16.1
 safetensors==0.4.0
-torch==2.0.1+cu117
+torch==2.1.0
 transformers==4.34.1
-xformers==0.0.22
+xformers==0.0.22.post7
 
 realesrgan==0.3.0
 basicsr>=1.4.2