Merge pull request #181 from hodanov/feature/dockerfile

Update pytorch.

app/Dockerfile

@@ -1,8 +1,8 @@
-FROM python:3.11.3-slim-bullseye
+FROM python:3.11.11-slim-bookworm
 COPY ./requirements.txt /
 RUN apt-get update \
     && apt-get install wget libgl1-mesa-glx libglib2.0-0 --no-install-recommends -y \
     && apt-get autoremove -y \
     && apt-get clean -y \
     && rm -rf /var/lib/apt/lists/* \
-    && pip install -r requirements.txt --extra-index-url https://download.pytorch.org/whl/cu121 --no-cache-dir
+    && pip install -r requirements.txt --extra-index-url https://download.pytorch.org/whl/cu126 --no-cache-dir

app/requirements.txt

@@ -1,11 +1,11 @@
 diffusers[torch]==0.32.2
 accelerate
 torch==2.6.0
-transformers==4.48.1
+transformers==4.48.2
 xformers==0.0.29.post2
 
 invisible_watermark # To help viewers identify the images as machine-generated.
-safetensors==0.5.2 # To store tensors safely.
+safetensors==0.5.0 # To store tensors safely.
 controlnet_aux
 
 Pillow