From 6c3a7b6b81c79e62b01e1c0015608cb5ee549ee8 Mon Sep 17 00:00:00 2001
From: Abdullah Barhoum <ka70911@gmail.com>
Date: Sun, 18 Sep 2022 11:49:37 +0200
Subject: [PATCH] lstein

---
 services/lstein/Dockerfile | 13 +++++++------
 services/lstein/mount.sh   |  6 +++++-
 2 files changed, 12 insertions(+), 7 deletions(-)

diff --git a/services/lstein/Dockerfile b/services/lstein/Dockerfile
index b4321d8..0ddef3e 100644
--- a/services/lstein/Dockerfile
+++ b/services/lstein/Dockerfile
@@ -6,7 +6,8 @@ SHELL ["/bin/bash", "-ceuxo", "pipefail"]
 
 ENV DEBIAN_FRONTEND=noninteractive
 
-RUN conda install python=3.8.5 && conda clean -a -y
+# now it requires python3.9
+# RUN conda install python=3.8.5 && conda clean -a -y
 RUN conda install pytorch==1.11.0 torchvision==0.12.0 cudatoolkit=11.3 -c pytorch && conda clean -a -y
 
 RUN apt-get update && apt install fonts-dejavu-core rsync gcc -y && apt-get clean
@@ -15,13 +16,13 @@ RUN apt-get update && apt install fonts-dejavu-core rsync gcc -y && apt-get clea
 RUN <<EOF
 git clone https://github.com/lstein/stable-diffusion.git
 cd stable-diffusion
-git reset --hard 751283a2de81bee4bb571fbabe4adb19f1d85b97
-conda env update --file environment.yaml -n base
-conda clean -a -y
+# git reset --hard 751283a2de81bee4bb571fbabe4adb19f1d85b97
+# conda env update --file environment.yaml -n base
+# conda clean -a -y
 EOF
 
 
-ARG BRANCH=development SHA=45af30f3a4c98b50c755717831c5fff75a3a8b43
+ARG BRANCH=development SHA=9bcb0dff968bd7732ca7fa61b7cb3f6e1c8a7943
 # ARG BRANCH=main SHA=89da371f4841f7e05da5a1672459d700c3920784
 RUN <<EOF
 cd stable-diffusion
@@ -32,7 +33,7 @@ conda env update --file environment.yaml -n base
 conda clean -a -y
 EOF
 
-RUN pip uninstall opencv-python -y && pip install --prefer-binary --upgrade --force-reinstall --no-cache-dir opencv-python-headless
+RUN pip uninstall opencv-python -y && pip install --prefer-binary --force-reinstall --no-cache-dir opencv-python-headless transformers==4.22
 
 COPY . /docker/
 RUN python3 /docker/info.py /stable-diffusion/static/dream_web/index.html && chmod +x /docker/mount.sh
diff --git a/services/lstein/mount.sh b/services/lstein/mount.sh
index ef36444..bae7a0b 100755
--- a/services/lstein/mount.sh
+++ b/services/lstein/mount.sh
@@ -15,12 +15,16 @@ if test -f /cache/models/GFPGANv1.3.pth; then
 fi
 
 # facexlib
-FACEX_WEIGHTS=/opt/conda/lib/python3.8/site-packages/facexlib/weights
+FACEX_WEIGHTS=/opt/conda/lib/python3.9/site-packages/facexlib/weights
 
 rm -rf "${FACEX_WEIGHTS}"
 mkdir -p /cache/weights
 ln -sf -T /cache/weights "${FACEX_WEIGHTS}"
 
+REALESRGAN_WEIGHTS=/opt/conda/lib/python3.9/site-packages/realesrgan/weights
+rm -rf "${REALESRGAN_WEIGHTS}"
+ln -sf -T /cache/weights "${REALESRGAN_WEIGHTS}"
+
 if "${PRELOAD}" == "true"; then
   python3 -u scripts/preload_models.py
 fi