From 19bcf71892d24de6d30c103b38ee61d8af2c6417 Mon Sep 17 00:00:00 2001
From: hodanov <1031hoda@gmail.com>
Date: Fri, 3 Nov 2023 17:33:42 +0900
Subject: [PATCH] Update xformers and torch.

---
 setup_files/Dockerfile       | 2 +-
 setup_files/requirements.txt | 4 ++--
 2 files changed, 3 insertions(+), 3 deletions(-)

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