diff --git a/services/hlky/Dockerfile b/services/hlky/Dockerfile
index 4d7eee9..8ddcd5f 100644
--- a/services/hlky/Dockerfile
+++ b/services/hlky/Dockerfile
@@ -16,7 +16,7 @@ RUN <<EOF
 git config --global http.postBuffer 1048576000
 git clone https://github.com/sd-webui/stable-diffusion-webui.git stable-diffusion
 cd stable-diffusion
-git reset --hard 7623a5734740025d79b710f3744bff9276e1467b
+git reset --hard 1a9c053cb7b6832695771db2555c0adc9b41e95f
 conda env update --file environment.yaml -n base
 conda clean -a -y
 EOF
@@ -24,8 +24,8 @@ EOF
 
 # 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 BRANCH=master  SHA=d0bb60a139d60e6c2b9be4e18e0e29a86aa5af59
-ARG BRANCH=dev SHA=1e7bdfe3f38a6dd37fc230f440ea1b0db0937240
+ARG BRANCH=master  SHA=1a9c053cb7b6832695771db2555c0adc9b41e95f
+# ARG BRANCH=dev SHA=1e7bdfe3f38a6dd37fc230f440ea1b0db0937240
 RUN <<EOF
 cd stable-diffusion
 git fetch
diff --git a/services/hlky/mount.sh b/services/hlky/mount.sh
index 95e7f13..c5499a7 100755
--- a/services/hlky/mount.sh
+++ b/services/hlky/mount.sh
@@ -9,11 +9,14 @@ ROOT=/stable-diffusion/src
 # cache
 MOUNTS["/root/.cache"]=/data/.cache
 # ui specific
-MOUNTS["${ROOT}/gfpgan/experiments/pretrained_models/GFPGANv1.3.pth"]=/data/GFPGAN/GFPGANv1.4.pth
-MOUNTS["${ROOT}/realesrgan/experiments/pretrained_models"]=/data/RealESRGAN
-MOUNTS["${ROOT}/latent-diffusion/experiments/pretrained_models"]=/data/LDSR
-# hacks
-MOUNTS["/stable-diffusion/gfpgan/weights"]=/data/.cache
+MOUNTS["${PWD}/models/realesrgan"]=/data/RealESRGAN
+MOUNTS["${PWD}/models/ldsr"]=/data/LDSR
+MOUNTS["${PWD}/models/custom"]=/data/StableDiffusion
+
+# hack
+MOUNTS["${PWD}/models/gfpgan/GFPGANv1.3.pth"]=/data/GFPGAN/GFPGANv1.4.pth
+MOUNTS["${PWD}/models/gfpgan/GFPGANv1.4.pth"]=/data/GFPGAN/GFPGANv1.4.pth
+
 
 for to_path in "${!MOUNTS[@]}"; do
   set -Eeuo pipefail