commit 412998e7980d10ba1d44e1b588618932d0e305c4
Author: Abdullah Barhoum <ka70911@gmail.com>
Date:   Sat Aug 27 12:38:12 2022 +0100

    Initial

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 0000000..c4638dc
Binary files /dev/null and b/README.md differ
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