From f097b266c2c90abe40f16f34c0ead93ac2135be3 Mon Sep 17 00:00:00 2001
From: Abdullah Barhoum <ka70911@gmail.com>
Date: Sun, 11 Sep 2022 05:02:58 +0200
Subject: [PATCH] Update hlky to dev branch

---
 docker-compose.yml |  1 +
 hlky/Dockerfile    | 13 ++++++++++---
 2 files changed, 11 insertions(+), 3 deletions(-)

diff --git a/docker-compose.yml b/docker-compose.yml
index cf9a3b4..32b9ec3 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -7,6 +7,7 @@ services:
       args:
         # You can choose any commit sha from https://github.com/hlky/stable-diffusion/commits/main
         # USE AT YOUR OWN RISK! otherwise just leave it empty.
+        BRANCH:
         WEBUI_SHA:
     ports:
       - "7860:7860"
diff --git a/hlky/Dockerfile b/hlky/Dockerfile
index 143f9d4..b06e788 100644
--- a/hlky/Dockerfile
+++ b/hlky/Dockerfile
@@ -23,9 +23,16 @@ RUN pip install -U --no-cache-dir pyperclip
 
 # Note: don't update the sha of previous versions because the install will take forever
 # instead, update the repo state in a later step
-ARG WEBUI_SHA=b9d97c9816251933d094f1dae43d2c631a07db7a
-RUN cd stable-diffusion && git pull && git reset --hard ${WEBUI_SHA} && \
-  conda env update --file environment.yaml --name base && conda clean -a -y
+ARG BRANCH=dev
+ARG WEBUI_SHA=be2ece06837e37d90181a17340c7e1aac91ba4fb
+RUN <<EOF
+cd stable-diffusion
+git fetch
+git checkout ${BRANCH}
+git reset --hard ${WEBUI_SHA}
+conda env update --file environment.yaml -n base
+conda clean -a -y
+EOF
 
 # Textual inversion
 RUN <<EOF