From 412998e7980d10ba1d44e1b588618932d0e305c4 Mon Sep 17 00:00:00 2001
From: Abdullah Barhoum <ka70911@gmail.com>
Date: Sat, 27 Aug 2022 12:38:12 +0100
Subject: [PATCH] Initial

---
 .gitignore         |   1 +
 README.md          | Bin 0 -> 4572 bytes
 build/Dockerfile   |  25 +++++++++++++++++++++++++
 cache/.gitignore   |   3 +++
 docker-compose.yml |  21 +++++++++++++++++++++
 models/.gitignore  |   4 ++++
 output/.gitignore  |   2 ++
 7 files changed, 56 insertions(+)
 create mode 100644 .gitignore
 create mode 100644 README.md
 create mode 100644 build/Dockerfile
 create mode 100644 cache/.gitignore
 create mode 100644 docker-compose.yml
 create mode 100644 models/.gitignore
 create mode 100644 output/.gitignore

diff --git a/.gitignore b/.gitignore
new file mode 100644
index 0000000..f42e49b
--- /dev/null
+++ b/.gitignore
@@ -0,0 +1 @@
+/dev
diff --git a/README.md b/README.md
new file mode 100644
index 0000000000000000000000000000000000000000..c4638dc9b92817b82a04de3318bdf871738f9084
GIT binary patch
literal 4572
zcmd6rT~A|G6oyy4H1W<$FPe}^B(Wci3?PXy2uLuQaRhWU#6VkUDYQM!X`#cvnO~|t
z?_TWgb4p=Ay)Y-Gr+e?U*1Nvf=C41OtYU{Yv8J`GZ|D5lv7U8!u5HX)W<%EAvD0Fu
z7xpu2+Z8LheP*AA&yiiQH)c0yeFT+gCw%92%j$^l%x+jaV_)YcP)OSG9q(Uk-~O;K
zS)K5VIJ0SWo^IfCf$Wymkek_n)eEp*!MV=4guM#)dF)l2BFC{WBd-~xpTQ^UP(fGW
ztZ+`d!aL@Dg3M8%7{a#_kdKg?abm#!kQ23;1r9lT4$mGu`ka=2P0m~ed9sA~-|&$|
zWL0V3;;$^{5=7%xYzEY0&Ph_V-zD#DIFIcs)~1{lMF!3ZzulsPus57Z$k5LBobxoC
z?+1;9P>39dG?FCwPTuo)*qmutWp~~?^0w?CooZ0#_^WPvCU43wIdT*ui)jB#&puDa
z7Wj_rCwpyMeDB+~t=v7&i>K#F-Nrrx?E7o*@y8{_+*-0FEPsflJtE`!0Q)Kn6|L7;
zT5pQ1>)@y8*>a>S(mZ4OA=?$#ic)=hw0oqUu{wccjrIKyLAU(!xD&@IG|Gf#piFTa
zvp;do;X5RkBWh{01lR9b@4>_4L%XNp(OVg_=e)<I$7+!>QHfIe#Y|8w%6OC;DGS6Q
zqEr?|i?xFtD#0qEP;MzNqAaZxdAB&P3epWyq@!#&4H@ipG;;1DU6}l@S3PC_`yfHy
zP{q7&p&cmh+gbQyl~XP8{HqXI8SCfZSLKY>J|@PMm7)@URXSy{Q!c8vk>g_-y*%Tu
zync-(#CwzXk9CsGP@TLArtY#roUK9~eW0o+EzA0!2MZrRivA2;d!eS)@T|1#6zb9x
z>(uQ#PMq1dwo43l$p0MMwW$HyRD~^SSJQrG=Qr|n-!APHG(U1;7rB=pdP}94;Pt*O
z2VLA=O-=~2d-=5B4fU)m{5v7K<rP_M&whZ<9@5^-XnN4ZE-6l3_$f+q@F+qQU7Zkf
z%0uPl{rEU8alQd*&A#R%=@-xFu17(3JN%u1@&MbYswdQSa4FWjV+v9Ip1<n&AtGqZ
z(Wf29whYQ>orA!&kHrs>9IX_4v}&C?ZH^_!*h5z9hxqFTf6PaXCX7WKPsqR3K&v_D
z`8IkVWUBWyiHiD6Mzw_dsdj3BK(>;do{rY%8LvD~(eebWLvSlrPi&1S|3B#c5)}HZ
zK1uh7b%-&fIC@KzsD5h}(==AD|MM~8*;TBd^VPnX(WfC3Gx)0B<{|1dbr$D4?}$UP
zSe!KP7AHz|v^->xrnx?qN}b5(H=jl$Z_#PGx~7V#`z!0T=2M?faq7bvHAd?`FBYXM
zH+)L))na!@z4MC~##15ocFFf0YWgl78K9x88Fyv3I5k&!^La+|q9z4R#r@Lx)IG)J
zlNTTC11LYglco9<xphFPYEqP@ZqN?ZzPMWSQju%KSgca>Rc6uor)S$gqkcGEi~OQZ
zl6I+Y=v>s!XZ=x#ZmlN|>Q|ht4%i<CTb9oh{ikkQ^k1C+s?^o8-^w!H1y3ob(y27<
z`D~dJt0|tFR9W4l{3aQ15PrK`A@kRX9o0Xtft&o^pf;%z3cWsWpj8jlgpsnOM>a^F
z=ZNkade^PUb6%A>&c`t?y!HvN=cKO-oAcEzF!?}d62FJ1l9av5g!?oplGVk_bL1(0
zh;hECvM8r3VG{6p(dTI)lg5&!sX%s6ElT%MpR%`u)s<b{nKkF-7o*~#tQN93&&DZD
zxv74lNhSHEjrH|kf>FTdbCF{c=5ifQH}hmczk$k^e%pw*lP(-}vq^PSGtlpF)M~^e
zO}!t+${+D4+BN^m5<>F$9QkkWruA@p&iGqrf_z19Yf(!Z#PtT&S_8*RK5MMD@Ag;i
t8!Wa)^mjPXU>aD5LeDx+gWX1;+JNI2D$Ti?oiuqRJF1`GGMOrC{sxne{Y(G=

literal 0
HcmV?d00001

diff --git a/build/Dockerfile b/build/Dockerfile
new file mode 100644
index 0000000..ccb6a57
--- /dev/null
+++ b/build/Dockerfile
@@ -0,0 +1,25 @@
+# syntax=docker/dockerfile:1
+
+FROM continuumio/miniconda3:4.12.0
+
+
+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 git clone https://github.com/hlky/stable-diffusion.git && cd stable-diffusion && git reset --hard 554bd068e6f2f6bc55449a67fe017ddd77090f28
+RUN conda env update --file stable-diffusion/environment.yaml --name base && conda clean -a -y
+
+
+
+ENV TRANSFORMERS_CACHE=/cache/transformers TORCH_HOME=/cache/torch CLI_ARGS=""
+
+WORKDIR /stable-diffusion
+
+EXPOSE 7860
+CMD ln -sf /models/model.ckpt /stable-diffusion/models/ldm/stable-diffusion-v1/model.ckpt && \
+  ln -sf /models/GFPGANv1.3.pth   /stable-diffusion/src/gfpgan/experiments/pretrained_models/GFPGANv1.3.pth && \
+  ln -sf /models/RealESRGAN_x4plus.pth   /stable-diffusion/src/realesrgan/experiments/pretrained_models/RealESRGAN_x4plus.pth && \
+  # force facexlib cache
+  mkdir -p /cache/weights/ && rm -rf /opt/conda/lib/python3.8/site-packages/facexlib/weights && \
+  ln -sf  /cache/weights/ /opt/conda/lib/python3.8/site-packages/facexlib/ && \
+  # run, -u to not buffer stdout / stderr
+  python3 -u scripts/webui.py ${CLI_ARGS}
diff --git a/cache/.gitignore b/cache/.gitignore
new file mode 100644
index 0000000..b4259b6
--- /dev/null
+++ b/cache/.gitignore
@@ -0,0 +1,3 @@
+/torch
+/transformers
+/weights
\ No newline at end of file
diff --git a/docker-compose.yml b/docker-compose.yml
new file mode 100644
index 0000000..53641d5
--- /dev/null
+++ b/docker-compose.yml
@@ -0,0 +1,21 @@
+version: '3.9'
+
+services:
+  model:
+    build: ./build/
+    restart: always
+    ports:
+      - "7860:7860"
+    volumes:
+      - ./cache:/cache
+      - ./output:/output
+      - ./models:/models
+    environment:
+      - CLI_ARGS=--outdir /output --save-metadata --ckpt /models/model.ckpt --no-progressbar-hiding
+    deploy:
+      resources:
+        reservations:
+          devices:
+              - driver: nvidia
+                device_ids: ['0']
+                capabilities: [gpu]
diff --git a/models/.gitignore b/models/.gitignore
new file mode 100644
index 0000000..60100b8
--- /dev/null
+++ b/models/.gitignore
@@ -0,0 +1,4 @@
+/model.ckpt
+/GFPGANv1.3.pth
+/RealESRGAN_x4plus.pth
+/RealESRGAN_x4plus_anime_6B.pth
diff --git a/output/.gitignore b/output/.gitignore
new file mode 100644
index 0000000..11b4bb1
--- /dev/null
+++ b/output/.gitignore
@@ -0,0 +1,2 @@
+/*
+!/.gitignore
\ No newline at end of file